Searched refs:bias_backprop (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.cu.cc | 103 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_Naive() argument
106 CudaAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNHWC_Naive()
113 T* bias_backprop, int32 bias_size, in BiasGradNCHW_Naive() argument
118 CudaAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNCHW_Naive()
127 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_SharedAtomics() argument
143 CudaAtomicAdd(bias_backprop + index, T(s_data[index])); in BiasGradNHWC_SharedAtomics()
149 T* bias_backprop, int32 batch, in BiasGradNCHW_SharedAtomics() argument
191 CudaAtomicAdd(bias_backprop + bias_index, T(data)); in BiasGradNCHW_SharedAtomics()
198 T* bias_backprop, int32 batch, int32 height, in compute() argument
220 d.stream()>>>(total_count, output_backprop, bias_backprop, in compute()
[all …]
|
| D | mkl_conv_grad_bias_ops.cc | 91 Tensor* bias_backprop = nullptr; in Compute() local
94 AllocateOutputSetMklShape(context, 0, &bias_backprop, output_shape, in Compute()
131 bias_backprop); in Compute()
139 static_cast<void*>(bias_backprop->flat<T>().data())), in Compute()
184 Tensor* bias_backprop) { in MklPrepareConvolutionOutputs()
202 static_cast<void*>(const_cast<T*>(bias_backprop->flat<T>().data())); in MklPrepareConvolutionOutputs()
|
| D | bias_op_gpu.h | 41 T* bias_backprop, int32 batch, int32 height, int32 width,
|