Searched refs:AddConstInput (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_to_space_nd_test.cc | 68 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in BatchToSpaceNDOpConstModel()
69 crops_ = AddConstInput(TensorType_INT32, crops, {2, 2}); in BatchToSpaceNDOpConstModel()
|
| D | pad_test.cc | 78 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
80 this->AddConstInput(GetTensorType<T1>(), {constant_values}, {1}); in PadV2OpConstModel()
96 this->AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadV2OpConstModel()
120 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
140 this->constant_values_ = this->AddConstInput( in PadV2OpDynamicModel()
|
| D | space_to_batch_nd_test.cc | 76 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in SpaceToBatchNDOpConstModel()
77 paddings_ = AddConstInput(TensorType_INT32, paddings, {2, 2}); in SpaceToBatchNDOpConstModel()
|
| D | reduce_test.cc | 64 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
93 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in SumOpConstModel()
122 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in ProdOpConstModel()
151 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MaxOpConstModel()
180 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MinOpConstModel()
209 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in AnyOpConstModel()
|
| D | split_test.cc | 35 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitOpModel()
|
| D | split_v_test.cc | 38 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitVOpModel()
|
| D | reshape_test.cc | 113 AddConstInput(TensorType_INT32, shape_data, shape_shape); in BuildWithConstantTensorShape()
|
| D | resize_nearest_neighbor_test.cc | 34 size_ = AddConstInput(TensorType_INT32, size_data, {2});
|
| D | transpose_test.cc | 182 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeOpConstModel()
|
| D | resize_bilinear_test.cc | 34 size_ = AddConstInput(TensorType_INT32, size_data, {2});
|
| D | test_util.h | 165 int AddConstInput(TensorType type, std::initializer_list<T> data, in AddConstInput() function
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 676 new_shape_ = AddConstInput<int>(TensorType_INT32, new_shape, in ReshapeOpModel()
804 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeSimpleModel()
1442 size_ = AddConstInput(TensorType_INT32, size_data, {2});
1573 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); in PadOpConstModel()
1623 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in SpaceToBatchNDOpConstModel()
1624 paddings_ = AddConstInput(TensorType_INT32, paddings, {2, 2}); in SpaceToBatchNDOpConstModel()
1689 begin_ = AddConstInput(TensorType_INT32, begin_data, begin_shape); in StridedSliceOpModel()
1690 end_ = AddConstInput(TensorType_INT32, end_data, end_shape); in StridedSliceOpModel()
1691 strides_ = AddConstInput(TensorType_INT32, strides_data, strides_shape); in StridedSliceOpModel()
3425 axis_ = AddConstInput(TensorType_INT32, axis, axis_shape); in MeanOpConstModel()
|