Searched refs:reverse_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_op.h | 29 const Eigen::array<bool, Dims>& reverse_dims, in operator()
31 output.device(d) = input.reverse(reverse_dims); in operator()
38 const Eigen::array<bool, 0>& reverse_dims,
|
| D | reverse_op.cc | 326 const Eigen::array<bool, DIM>& reverse_dims, \
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReverse.h | 65 const XprType& expr, const ReverseDimensions& reverse_dims)
66 : m_xpr(expr), m_reverse_dims(reverse_dims) { }
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 270 auto reverse_dims = op.input(1); in ReverseGrad() local
271 grad_outputs->push_back(Reverse(scope, grad_inputs[0], reverse_dims)); in ReverseGrad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 895 reverse_dims = op.inputs[1]
896 return gen_array_ops.reverse(grad, reverse_dims), None
|