Home
last modified time | relevance | path

Searched refs:AddConstInput (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbatch_to_space_nd_test.cc68 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in BatchToSpaceNDOpConstModel()
69 crops_ = AddConstInput(TensorType_INT32, crops, {2, 2}); in BatchToSpaceNDOpConstModel()
Dpad_test.cc78 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()
Dspace_to_batch_nd_test.cc76 block_shape_ = AddConstInput(TensorType_INT32, block_shape, {2}); in SpaceToBatchNDOpConstModel()
77 paddings_ = AddConstInput(TensorType_INT32, paddings, {2, 2}); in SpaceToBatchNDOpConstModel()
Dreduce_test.cc64 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()
Dsplit_test.cc35 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitOpModel()
Dsplit_v_test.cc38 axis_ = AddConstInput(TensorType_INT32, {axis}, {1}); in SplitVOpModel()
Dreshape_test.cc113 AddConstInput(TensorType_INT32, shape_data, shape_shape); in BuildWithConstantTensorShape()
Dresize_nearest_neighbor_test.cc34 size_ = AddConstInput(TensorType_INT32, size_data, {2});
Dtranspose_test.cc182 perm_ = AddConstInput(TensorType_INT32, perm, perm_shape); in TransposeOpConstModel()
Dresize_bilinear_test.cc34 size_ = AddConstInput(TensorType_INT32, size_data, {2});
Dtest_util.h165 int AddConstInput(TensorType type, std::initializer_list<T> data, in AddConstInput() function
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc676 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()