Searched defs:output_backprop (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.cu.cc | 104 const T* __restrict__ output_backprop, in BiasGradNHWC_Naive()
116 const T* __restrict__ output_backprop, in BiasGradNCHW_Naive()
128 int32 nthreads, const T* __restrict__ output_backprop, in BiasGradNHWC_SharedAtomics()
152 const T* __restrict__ output_backprop, T* __restrict__ bias_backprop, in BiasGradNCHW_SharedAtomics()
212 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute()
|
| D | bias_op.cc | 231 const Tensor& output_backprop = context->input(0); in Compute() local
459 const Tensor& output_backprop, int32 batch, in ComputeWithCustomKernel()
469 const Tensor& output_backprop, int32 batch, in ComputeWithReduceSum()
502 const Tensor& output_backprop = context->input(0); in Compute() local
|
| D | cudnn_rnn_ops.cc | 855 const Tensor* output_backprop, const Tensor* output_h_backprop, in DoBackward()
1671 Tensor output_backprop; in MaybeAutoTune() local
1886 const Tensor* output_backprop = nullptr; in ComputeImpl() local
1945 const Tensor** output_backprop, const Tensor** output_h_backprop, in ExtractBackwardInputs()
|
| D | nn_ops_test.cc | 955 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 | 1522 def _testMaxPoolGradDirect(self, input_data, output_backprop, argument
|