Home
last modified time | relevance | path

Searched defs:paddings (Results 1 – 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/python/kernel_tests/
Dspacetobatch_op_test.py34 def space_to_batch_direct(input_array, block_shape, paddings): argument
103 def _testPad(self, inputs, paddings, block_size, outputs): argument
200 def _testPad(self, inputs, block_shape, paddings, outputs): argument
214 def _testDirect(self, input_shape, block_shape, paddings): argument
422 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
431 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
449 def _testShape(self, input_shape, block_shape, paddings, error): argument
527 def _checkGrad(self, x, paddings, block_size): argument
586 def _checkGrad(self, x, block_shape, paddings): argument
603 def _compare(self, input_shape, block_shape, paddings): argument
[all …]
Dpad_op_test.py35 def _npPad(self, inp, paddings, mode, constant_values=0): argument
87 def _testPad(self, np_inputs, paddings, mode, constant_values): argument
116 def _testAll(self, np_inputs, paddings, constant_values): argument
Dbatchtospace_op_test.py139 def _testStaticShape(self, input_shape, block_shape, paddings, error): argument
148 def _testDynamicShape(self, input_shape, block_shape, paddings): argument
167 def _testShape(self, input_shape, block_shape, paddings, error): argument
/external/tensorflow/tensorflow/compiler/tests/
Dspacetobatch_op_test.py30 def space_to_batch_direct(input_array, block_shape, paddings): argument
74 def _testPad(self, inputs, paddings, block_size, outputs): argument
155 def _testPad(self, inputs, block_shape, paddings, outputs): argument
187 def _testDirect(self, input_shape, block_shape, paddings): argument
Dbinary_ops_test.py1239 mirror_pad = lambda t, paddings: array_ops.pad(t, paddings, "SYMMETRIC") argument
1270 mirror_pad = lambda t, paddings: array_ops.pad(t, paddings, "REFLECT") argument
1344 mirror_pad_grad = lambda t, paddings: gen_array_ops.mirror_pad_grad( argument
1360 mirror_pad_grad = lambda t, paddings: gen_array_ops.mirror_pad_grad( argument
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dpad_tester.h44 inline PadTester& InputPrePaddings(std::initializer_list<int32_t> paddings) { in InputPrePaddings()
57 inline PadTester& InputPostPaddings(std::initializer_list<int32_t> paddings) { in InputPostPaddings()
Dpad_tester.cc123 std::vector<int32_t> paddings(InputPrePaddings().size() + in CreateTfLiteModel() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/
Dops_defs.py52 def _composite_mirror_pad(input_, paddings, mode): argument
102 def _composite_mirror_pad_grad(input_, paddings, mode): argument
/external/eigen/unsupported/test/
Dcxx11_tensor_padding.cpp22 array<std::pair<ptrdiff_t, ptrdiff_t>, 4> paddings; in test_simple_padding() local
57 array<std::pair<ptrdiff_t, ptrdiff_t>, 4> paddings; in test_padded_expr() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc27 const xla::Literal& paddings) { in SpaceToBatch()
156 xla::Literal paddings; in Compile() local
178 xla::Literal paddings; in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dpad_op.cc79 typename TTypes<Tpadding>::ConstMatrix paddings = in1.matrix<Tpadding>(); in Compute() local
156 typename TTypes<Tpadding>::ConstMatrix paddings = in CollapseAdjacentNonPaddedDimensions() local
205 typename TTypes<Tpadding>::ConstMatrix paddings, in OperateWithVariableRank()
242 typename TTypes<Tpadding>::ConstMatrix paddings, T pad_value, in Operate()
Dspacetobatch_benchmark_test.cc26 DataType dtype, const std::vector<std::pair<int, int>>& paddings) { in ConstructSpaceToBatchGraph() argument
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dpad_test.cc25 std::initializer_list<int> paddings, in PadOpConstModel()
41 void SetPaddings(std::initializer_list<int> paddings) { in SetPaddings()
Dmirror_pad_test.cc26 std::initializer_list<int> paddings, in MirrorPadOpModel()
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad_test.cc269 auto paddings = Const(scope_, {{1, 1}, {2, 2}}); in TEST_F() local
278 auto paddings = Const(scope_, {{1, 1}, {1, 1}}); in TEST_F() local
288 auto paddings = Const(scope_, {{0, 0}, {2, 0}}); in TEST_F() local
297 auto paddings = Const(scope_, {{1, 1}, {1, 1}}); in TEST_F() local
307 auto paddings = Const(scope_, {{0, 0}, {2, 0}}); in TEST_F() local
332 auto paddings = Const(scope_, {{1, 1}, {2, 2}}); in TEST_F() local
341 auto paddings = Const(scope_, {{1, 1}, {2, 2}}); in TEST_F() local
350 auto paddings = Const(scope_, {{1, 1}, {2, 2}}); in TEST_F() local
359 auto paddings = Const(scope_, {{1, 1}, {2, 2}}); in TEST_F() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops_test.cc1387 Tensor paddings = test::AsTensor<int32>({4, 2, 2, 4}, {{2, 2}}); in TEST() local
1431 Tensor paddings = test::AsTensor<int32>({1, 1, 0, 1}, {{2, 2}}); in TEST() local
1439 Tensor paddings = test::AsTensor<int32>({1, 1, 0, 0}, {{2, 2}}); in TEST() local
1461 Tensor paddings = test::AsTensor<int32>({0, -1, 0, 0}, {{2, 2}}); in TEST() local
1471 Tensor paddings = test::AsTensor<int32>({0, 0, 0, 0}, {{2, 2}}); in TEST() local
1481 Tensor paddings = test::AsTensor<int32>({}); in TEST() local
1592 Tensor paddings = test::AsTensor<int32>({0, -1, 0, 0}, {{2, 2}}); in TEST() local
/external/tensorflow/tensorflow/core/kernels/image/
Dmirror_pad_op.cc85 typename TTypes<Tpaddings>::ConstMatrix paddings = in1.matrix<Tpaddings>(); in Compute() local
296 typename TTypes<Tpaddings>::ConstMatrix paddings = in1.matrix<Tpaddings>(); in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddilated_conv.h244 auto paddings = stb_paddings_attr.getValues<IntegerAttr>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpad.cc54 const TfLiteTensor* paddings = GetInput(context, node, /*index=*/1); in Prepare() local
/external/eigen/bench/tensors/
Dtensor_benchmarks.h183 Eigen::type2indexpair<2, 1> > paddings; in padding() local
185 Eigen::array<Eigen::IndexPair<TensorIndex>, 2> paddings; in padding() local
/external/tensorflow/tensorflow/lite/kernels/
Dpad_test.cc49 void SetPaddings(std::initializer_list<int> paddings) { in SetPaddings()
78 std::initializer_list<int> paddings, T1 constant_values, in PadV2OpConstModel()
95 std::initializer_list<int> paddings, in PadV2OpConstModel()
121 std::initializer_list<int> paddings, in PadOpConstModel()
Dspace_to_batch_nd.cc46 const TfLiteTensor* paddings; member
/external/skia/tests/
DSwizzlerTest.cpp75 const uint32_t paddings[] = { 0, 4 }; in DEF_TEST() local
/external/skqp/tests/
DSwizzlerTest.cpp75 const uint32_t paddings[] = { 0, 4 }; in DEF_TEST() local
/external/tensorflow/tensorflow/stream_executor/
Ddnn.proto105 repeated int64 paddings = 1; field

123