Searched refs:reverse_dims (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dreverse_op.h29 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 Dconv_2d.h36 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 Dreverse_op.cc329 const Eigen::array<bool, DIM>& reverse_dims, \
H A Dconv_grad_filter_ops.cc988 const Eigen::array<bool, 4>& reverse_dims, \
H A Dconv_grad_input_ops.cc1061 const Eigen::array<bool, 4>& reverse_dims, \
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorReverse.h65 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 Darray_grad.cc243 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 Darray_grad.py629 reverse_dims = op.inputs[1]
631 return gen_array_ops._reverse(grad, reverse_dims), None

Completed in 353 milliseconds