Searched defs:output_backprop (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.cu.cc | 102 __global__ void BiasGradNHWC_Naive(int32 nthreads, const T* output_backprop, in BiasGradNHWC_Naive()
112 __global__ void BiasGradNCHW_Naive(int32 nthreads, const T* output_backprop, in BiasGradNCHW_Naive()
126 const T* output_backprop, in BiasGradNHWC_SharedAtomics()
148 __global__ void BiasGradNCHW_SharedAtomics(const T* output_backprop, in BiasGradNCHW_SharedAtomics()
197 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute()
|
| D | bias_op.cc | 246 const Tensor& output_backprop = context->input(0); in Compute() local
491 const Tensor& output_backprop, int32 batch, in ComputeWithCustomKernel()
501 const Tensor& output_backprop, int32 batch, in ComputeWithReduceSum()
534 const Tensor& output_backprop = context->input(0); in Compute() local
|
| D | cudnn_rnn_ops.cc | 796 const Tensor* output_backprop, const Tensor* output_h_backprop, in DoBackward()
1485 Tensor output_backprop; in MaybeAutoTune() local
1687 const Tensor* output_backprop = nullptr; in ComputeImpl() local
1746 const Tensor** output_backprop, const Tensor** output_h_backprop, in ExtractBackwardInputs()
|
| D | nn_ops_test.cc | 932 Tensor output_backprop(DT_FLOAT, output_shape); in BM_AvgPoolBk() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 85 NodeDef* output_backprop = graph.add_node(); in TEST() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 1274 def _testMaxPoolGradDirect(self, input_data, output_backprop, argument
|