Searched refs:grad_in (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 959 const Tensor& grad_in, const Tensor& argmax, in launch()
964 auto shard = [&grad_in, &argmax, &grad_out, include_batch_in_index]( in launch()
969 const int64 input_size_per_batch = grad_in.NumElements() / batch_size; in launch()
974 auto grad_in_flat = grad_in.flat<T>(); in launch()
1035 const Tensor& grad_in = context->input(1); in Compute() local
1051 context, params, grad_in, argmax, grad_out, include_batch_in_index_); in Compute()
1088 const Tensor& grad_in = context->input(1); in Compute() local
1105 context, params, grad_in, argmax, grad_out, include_batch_in_index_); in Compute()
1340 const Tensor& grad_in, const Tensor& argmax, in launch()
1350 output_size, input_size, grad_in.flat<T>().data(), in launch()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 790 grad_in = constant_op.constant(tensor_output, shape=output_shape)
791 out_op = gen_nn_ops.max_pool_grad_with_argmax(t, grad_in, argmax, ksize,
799 grad_in = constant_op.constant(tensor_output, shape=output_shape)
800 out_op = gen_nn_ops.max_pool_grad(t, orig_out, grad_in, ksize, strides,
819 grad_in = constant_op.constant(tensor_input, shape=input_shape)
821 t, grad_in, argmax, ksize, strides, padding)
828 grad_in = constant_op.constant(tensor_input, shape=input_shape)
829 out_op = gen_nn_ops.max_pool_grad_grad(t, orig_out, grad_in, ksize,
|