Searched refs:padding_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | mirror_pad_op.h | 68 const XprType& expr, const PaddingDimensions& padding_dims, Index offset)
69 : xpr_(expr), padding_dims_(padding_dims), offset_(offset) {}
351 Eigen::array<Eigen::IndexPair<int32>, Dims> padding_dims;
354 padding_dims[i] = Eigen::IndexPair<int32>(padding(i, 0), padding(i, 1));
357 output.device(device) = MirrorPadOp(input, padding_dims, offset);
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorPadding.h | 63 …_INLINE TensorPaddingOp(const XprType& expr, const PaddingDimensions& padding_dims, const Scalar p…
64 : m_xpr(expr), m_padding_dims(padding_dims), m_padding_value(padding_value) {}
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util.cc | 332 int padding_dims = padding.size(); in BuildPadLoopBodySubgraph() local
333 CreateConstantInt32Tensor(subgraph, kConstPadding, {1, padding_dims}, in BuildPadLoopBodySubgraph()
|