Home
last modified time | relevance | path

Searched defs:output_backprop (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbias_op_gpu.cu.cc107 const T* __restrict__ output_backprop, in BiasGradNHWC_Naive()
119 const T* __restrict__ output_backprop, in BiasGradNCHW_Naive()
131 int32 nthreads, const T* __restrict__ output_backprop, in BiasGradNHWC_SharedAtomics()
155 const T* __restrict__ output_backprop, T* __restrict__ bias_backprop, in BiasGradNCHW_SharedAtomics()
215 void BiasGradGPU<T>::compute(const GPUDevice& d, const T* output_backprop, in compute()
H A Dbias_op.cc173 const Tensor& output_backprop = context->input(0); in Compute() local
400 const Tensor& output_backprop, int32_t batch, in ComputeWithCustomKernel()
410 const Tensor& output_backprop, int32_t batch, in ComputeWithReduceSum()
443 const Tensor& output_backprop = context->input(0); in Compute() local
H A Dcudnn_rnn_ops.cc878 const Tensor* output_backprop, const Tensor* output_h_backprop, in DoBackward()
1715 Tensor output_backprop; in MaybeAutotune() local
1930 const Tensor* output_backprop = nullptr; in ComputeImpl() local
1989 const Tensor** output_backprop, const Tensor** output_h_backprop, in ExtractBackwardInputs()
H A Dnn_ops_test.cc970 Tensor output_backprop(DT_FLOAT, output_shape); in BM_AvgPoolBk() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dutils_test.cc85 NodeDef* output_backprop = graph.add_node(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dpooling_ops_test.py1564 def _testMaxPoolGradDirect(self, input_data, output_backprop, argument