Searched refs:reverse_dims (Results 1 - 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | reverse_op.h | 29 const Eigen::array<bool, Dims>& reverse_dims, 31 output.device(d) = input.reverse(reverse_dims); 38 const Eigen::array<bool, 0>& reverse_dims, 28 operator ()(const Device& d, typename TTypes<T, Dims>::ConstTensor input, const Eigen::array<bool, Dims>& reverse_dims, typename TTypes<T, Dims>::Tensor output) argument 37 operator ()(const Device& d, typename TTypes<T, 0>::ConstTensor input, const Eigen::array<bool, 0>& reverse_dims, typename TTypes<T, 0>::Tensor output) argument
|
H A D | conv_2d.h | 36 const Eigen::array<bool, Dims>& reverse_dims, 38 output.device(d) = input.shuffle(order).reverse(reverse_dims); 33 operator ()(const Device& d, typename TTypes<T, Dims, IndexType>::ConstTensor input, const Eigen::DSizes<IndexType, Dims>& order, const Eigen::array<bool, Dims>& reverse_dims, typename TTypes<T, Dims, IndexType>::Tensor output) argument
|
H A D | reverse_op.cc | 329 const Eigen::array<bool, DIM>& reverse_dims, \
|
H A D | conv_grad_filter_ops.cc | 988 const Eigen::array<bool, 4>& reverse_dims, \
|
H A D | conv_grad_input_ops.cc | 1061 const Eigen::array<bool, 4>& reverse_dims, \
|
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorReverse.h | 65 const XprType& expr, const ReverseDimensions& reverse_dims) 66 : m_xpr(expr), m_reverse_dims(reverse_dims) { } 64 TensorReverseOp( const XprType& expr, const ReverseDimensions& reverse_dims) argument
|
/external/tensorflow/tensorflow/cc/gradients/ |
H A D | array_grad.cc | 243 auto reverse_dims = op.input(1); local 244 grad_outputs->push_back(Reverse(scope, grad_inputs[0], reverse_dims));
|
/external/tensorflow/tensorflow/python/ops/ |
H A D | array_grad.py | 629 reverse_dims = op.inputs[1] 631 return gen_array_ops._reverse(grad, reverse_dims), None
|
Completed in 353 milliseconds