Searched refs:paddings_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pad_test.cc | 73 std::initializer_list<int> paddings_shape, in PadV2OpConstModel() argument
78 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
90 std::initializer_list<int> paddings_shape, in PadV2OpConstModel() argument
96 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
116 std::initializer_list<int> paddings_shape, in PadOpConstModel() argument
120 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
135 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel() argument
146 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
149 std::initializer_list<int> paddings_shape, in PadV2OpDynamicModel() argument
159 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
[all …]
|
| /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/core/kernels/ |
| D | nn_ops_test.cc | 165 TensorShape paddings_shape({4, 2}); in BM_ConvFloat() local
166 Tensor paddings_tensor(DT_INT32, paddings_shape); in BM_ConvFloat()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2079 ShapeHandle paddings_shape, in SpaceToBatchShapeHelper() argument
2096 c->Merge(paddings_shape, c->Matrix(num_block_dims, 2), &paddings_shape)); in SpaceToBatchShapeHelper()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 1569 std::initializer_list<int> paddings_shape, in PadOpConstModel() argument
1573 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
|