Searched defs:input_backprop (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 881 Tensor* input_backprop, Tensor* input_h_backprop, Tensor* input_c_backprop, in DoBackward()
1718 Tensor input_backprop; in MaybeAutotune() local
1940 Tensor* input_backprop = nullptr; in ComputeImpl() local
2047 Tensor** input_backprop, Tensor** input_h_backprop, in AllocateOutputs()
|
| H A D | cudnn_pooling_gpu.cc | 134 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
| H A D | quantize_and_dequantize_op.cc | 160 Tensor* input_backprop = nullptr; in Compute() local
|
| H A D | pooling_ops_common.cc | 452 Tensor* input_backprop = nullptr; in Compute() local
|
| H A D | pooling_ops_3d.cc | 348 Tensor* input_backprop; in Compute() local
|
| H A D | maxpooling_op.cc | 65 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
|