Searched defs:paddings_shape (Results 1 – 13 of 13) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | SpaceToBatchFixture.h | 43 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
63 …sorShape input_shape, const TensorShape &block_shape_shape, const TensorShape &paddings_shape, Ten… in compute_target()
115 … TensorShape &input_shape, const TensorShape &block_shape_shape, const TensorShape &paddings_shape, in compute_reference()
144 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
156 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pad_test.cc | 77 std::initializer_list<int> paddings_shape, in PadV2OpConstModel()
94 std::initializer_list<int> paddings_shape, in PadV2OpConstModel()
120 std::initializer_list<int> paddings_shape, in PadOpConstModel()
139 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel()
153 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel()
177 std::initializer_list<int> paddings_shape, in PadOpDynamicModel()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | pad_test.cc | 26 std::initializer_list<int> paddings_shape, in PadOpConstModel()
|
| D | mirror_pad_test.cc | 27 std::initializer_list<int> paddings_shape, in MirrorPadOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | pad_tester.cc | 138 const std::array<int32_t, 2> paddings_shape{ in CreateTfLiteModel() local
|
| D | quantized_pad_tester.cc | 146 const std::array<int32_t, 2> paddings_shape{ in CreateTfLiteModel() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | conv_spmd_expander.cc | 124 llvm::SmallVector<int64_t, 4> paddings_shape(input_tensor_shape.begin(), in PadInputOnUnshardedDim() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 101 TensorShape paddings_shape({4, 2}); in TEST() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_builder.cc | 133 static std::vector<int> paddings_shape = {1, 1, 2, 2}; in BuildDilatedDwConv() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2477 std::initializer_list<int> paddings_shape, in PadOpConstModel()
5083 std::initializer_list<int> paddings_shape, in PadV2OpConstModel()
5100 std::initializer_list<int> paddings_shape, in PadV2OpConstModel()
5122 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel()
5136 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 2253 auto paddings_shape = ops::Placeholder::Shape({5, 2}); in RunTest() local
2331 auto paddings_shape = ops::Placeholder::Shape({5, 2}); in RunTest() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nn_ops_test.cc | 171 TensorShape paddings_shape({4, 2}); in BM_ConvFloat() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2269 ShapeHandle paddings_shape, in SpaceToBatchShapeHelper()
|