Searched refs:grad_out (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 960 Tensor* grad_out, const bool include_batch_in_index) { in launch()
964 auto shard = [&grad_in, &argmax, &grad_out, include_batch_in_index]( in launch()
967 GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
968 const int64 output_size_per_batch = grad_out->NumElements() / batch_size; in launch()
972 auto grad_out_flat = grad_out->flat<T>(); in launch()
998 const int64 batch_size = GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
999 const int64 shard_cost = grad_out->NumElements() / batch_size; in launch()
1046 Tensor* grad_out = nullptr; in Compute() local
1048 {0}, 0, out_shape, &grad_out)); in Compute()
1051 context, params, grad_in, argmax, grad_out, include_batch_in_index_); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 794 grad_out = array_ops.reshape(jac, (rows_in, cols_in, batch_size, channels))
795 grad_out = array_ops.transpose(grad_out, (2, 0, 1, 3))
797 return [grad_out]
855 grad_out = array_ops.reshape(
857 grad_out = array_ops.transpose(grad_out, (3, 0, 1, 2, 4))
859 return [grad_out]
|