Searched defs:input_backprop (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_pooling_gpu.cc | 139 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
D | quantize_and_dequantize_op.cc | 154 Tensor* input_backprop = nullptr; in Compute() local
|
D | cudnn_rnn_ops.cc | 858 Tensor* input_backprop, Tensor* input_h_backprop, Tensor* input_c_backprop, in DoBackward() 1674 Tensor input_backprop; in MaybeAutoTune() local 1896 Tensor* input_backprop = nullptr; in ComputeImpl() local 2003 Tensor** input_backprop, Tensor** input_h_backprop, in AllocateOutputs()
|
D | pooling_ops_common.cc | 453 Tensor* input_backprop = nullptr; in Compute() local
|
D | pooling_ops_3d.cc | 347 Tensor* input_backprop; in Compute() local
|
D | maxpooling_op.cc | 64 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
|