HomeSort by relevance Sort by last modified time
    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