OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:bias_index
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
bias_op_gpu.cu.cc
154
int32
bias_index
= blockIdx.x % bias_size;
local
163
(batch * bias_size +
bias_index
) * image_size + image_offset);
182
CudaAtomicAdd(bias_backprop +
bias_index
, T(data));
mkl_conv_ops.h
195
virtual inline void GetBiasSizeInMklOrder(size_t
bias_index
,
197
const Tensor& bias = MklGetInput(context_,
bias_index
);
Completed in 77 milliseconds