Searched refs:reverse_dims (Results 1 – 8 of 8) 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 | conv_2d.h | 36 const Eigen::array<bool, Dims>& reverse_dims, in operator()
38 output.device(d) = input.shuffle(order).reverse(reverse_dims); in operator()
|
| D | reverse_op.cc | 329 const Eigen::array<bool, DIM>& reverse_dims, \
|
| D | conv_grad_filter_ops.cc | 914 const Eigen::array<bool, 4>& reverse_dims, \
|
| D | conv_grad_input_ops.cc | 1057 const Eigen::array<bool, 4>& 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 | 261 auto reverse_dims = op.input(1); in ReverseGrad() local
262 grad_outputs->push_back(Reverse(scope, grad_inputs[0], reverse_dims)); in ReverseGrad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 640 reverse_dims = op.inputs[1]
641 return gen_array_ops.reverse(grad, reverse_dims), None
|