| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pad_test.cc | 77 std::initializer_list<int> paddings_shape, in PadV2OpConstModel() argument
82 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
94 std::initializer_list<int> paddings_shape, in PadV2OpConstModel() argument
100 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
120 std::initializer_list<int> paddings_shape, in PadOpConstModel() argument
124 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
139 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel() argument
150 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
153 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel() argument
163 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | pad_test.cc | 24 std::initializer_list<int> paddings_shape, in PadOpConstModel() argument
28 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
|
| D | mirror_pad_test.cc | 25 std::initializer_list<int> paddings_shape, in MirrorPadOpModel() argument
30 AddConstInput(TensorType_INT32, paddings, 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
146 builder.CreateVector<int32_t>(paddings_shape.data(), in CreateTfLiteModel()
147 paddings_shape.size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_builder.cc | 124 static std::vector<int> paddings_shape = {1, 1, 2, 2}; in PopulateSubGraph() local
244 paddings_shape.data(), in PopulateSubGraph()
248 paddings_shape.data(), in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils_test.cc | 101 TensorShape paddings_shape({4, 2}); in TEST() local
102 Tensor paddings_tensor(DT_INT32, paddings_shape); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2400 std::initializer_list<int> paddings_shape, in PadOpConstModel() argument
2404 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
5006 std::initializer_list<int> paddings_shape, in PadV2OpConstModel() argument
5011 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
5023 std::initializer_list<int> paddings_shape, in PadV2OpConstModel() argument
5029 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
5045 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel() argument
5056 this->BuildInterpreterWithNNAPI({input.shape, paddings_shape}); in PadV2OpDynamicModel()
5059 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel() argument
5069 this->BuildInterpreterWithNNAPI({input.shape, paddings_shape}); in PadV2OpDynamicModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nn_ops_test.cc | 167 TensorShape paddings_shape({4, 2}); in BM_ConvFloat() local
168 Tensor paddings_tensor(DT_INT32, paddings_shape); in BM_ConvFloat()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2236 ShapeHandle paddings_shape, in SpaceToBatchShapeHelper() argument
2253 c->Merge(paddings_shape, c->Matrix(num_block_dims, 2), &paddings_shape)); in SpaceToBatchShapeHelper()
|