Home
last modified time | relevance | path

Searched refs:SetTensorParametersReadWrite (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc69 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
74 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
112 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
117 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
146 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
151 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
153 interpreter.SetTensorParametersReadWrite(3, kTfLiteFloat32, "d", {3}, in TEST_P()
193 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
198 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
200 interpreter.SetTensorParametersReadWrite(3, kTfLiteFloat32, "d", {3}, in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dvariable_ops_test.cc60 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", 0, in ConstructInvalidGraph()
62 interpreter_->SetTensorParametersReadWrite(1, kTfLiteResource, "", 0, in ConstructInvalidGraph()
64 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", 0, in ConstructInvalidGraph()
95 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", 0, in ConstructGraph()
97 interpreter_->SetTensorParametersReadWrite(1, kTfLiteResource, "", 0, in ConstructGraph()
99 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", 0, in ConstructGraph()
126 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {2, 2}, in ConstructGraphWithKnownShape()
128 interpreter_->SetTensorParametersReadWrite(1, kTfLiteResource, "", 0, in ConstructGraphWithKnownShape()
130 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {2, 2}, in ConstructGraphWithKnownShape()
Dhashtable_ops_test.cc353 interpreter_->SetTensorParametersReadWrite( in AddTensors()
357 interpreter_->SetTensorParametersReadWrite(kKeyTensorId, key_type_, "", in AddTensors()
362 interpreter_->SetTensorParametersReadWrite( in AddTensors()
367 interpreter_->SetTensorParametersReadWrite( in AddTensors()
372 interpreter_->SetTensorParametersReadWrite( in AddTensors()
377 interpreter_->SetTensorParametersReadWrite(kSizeTensorId, kTfLiteInt64, "", in AddTensors()
381 interpreter_->SetTensorParametersReadWrite( in AddTensors()
386 interpreter_->SetTensorParametersReadWrite( in AddTensors()
390 interpreter_->SetTensorParametersReadWrite( in AddTensors()
395 interpreter_->SetTensorParametersReadWrite( in AddTensors()
[all …]
Dsubgraph_test_util.cc77 ASSERT_EQ(subgraph->SetTensorParametersReadWrite(tensor_index, type, "", 0, in SetupTensor()
/external/armnn/delegate/classic/src/test/
DArmnnClassicDelegateTest.cpp30 …tfLiteInterpreter->SetTensorParametersReadWrite(0, kTfLiteFloat32, "input1", {1,2,2,1}, TfLiteQuan…
31 …tfLiteInterpreter->SetTensorParametersReadWrite(1, kTfLiteFloat32, "input2", {1,2,2,1}, TfLiteQuan…
32 …tfLiteInterpreter->SetTensorParametersReadWrite(2, kTfLiteFloat32, "output", {1,2,2,1}, TfLiteQuan…
68 …tfLiteInterpreter->SetTensorParametersReadWrite(0, kTfLiteFloat32, "input1", {1,2,2,1}, TfLiteQuan…
69 …tfLiteInterpreter->SetTensorParametersReadWrite(1, kTfLiteFloat32, "input2", {1,2,2,1}, TfLiteQuan…
70 …tfLiteInterpreter->SetTensorParametersReadWrite(2, kTfLiteFloat32, "output", {1,2,2,1}, TfLiteQuan…
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc99 interpreter.SetTensorParametersReadWrite(tensor_index, kTfLiteFloat32, "", in TEST()
119 interpreter.SetTensorParametersReadWrite(tensor_index, kTfLiteInt8, "", in TEST()
212 interpreter.SetTensorParametersReadWrite(0, test.type, "", {3}, quant); in TEST()
213 interpreter.SetTensorParametersReadWrite(1, test.type, "", {4}, quant); in TEST()
252 ASSERT_EQ(interpreter.SetTensorParametersReadWrite(0, tensor_type, "", {3}, in TEST()
302 interpreter.SetTensorParametersReadWrite(0, test.type, "", {3}, quant), in TEST()
336 interpreter.SetTensorParametersReadWrite(i, test.type, "", {2 * i + 1}, in TEST()
357 interpreter.SetTensorParametersReadWrite(static_cast<int>(i), kTfLiteUInt8, in TEST()
386 ASSERT_EQ(interpreter.SetTensorParametersReadWrite( in TEST()
404 ASSERT_EQ(interpreter.SetTensorParametersReadWrite( in TEST()
[all …]
Dinterpreter.cc258 TfLiteStatus Interpreter::SetTensorParametersReadWrite( in SetTensorParametersReadWrite() function in tflite::Interpreter
262 return primary_subgraph().SetTensorParametersReadWrite( in SetTensorParametersReadWrite()
277 TfLiteStatus Interpreter::SetTensorParametersReadWrite( in SetTensorParametersReadWrite() function in tflite::Interpreter
282 return primary_subgraph().SetTensorParametersReadWrite( in SetTensorParametersReadWrite()
Dinterpreter.h182 TfLiteStatus SetTensorParametersReadWrite(int tensor_index, TfLiteType type,
189 inline TfLiteStatus SetTensorParametersReadWrite(
200 return SetTensorParametersReadWrite(
204 TfLiteStatus SetTensorParametersReadWrite(
Dinterpreter_builder.cc657 if (subgraph->SetTensorParametersReadWrite( in ParseTensors()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc98 subgraph->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
100 subgraph->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
102 subgraph->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
104 subgraph->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
106 subgraph->SetTensorParametersReadWrite(4, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
393 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {1}, quant); in SetUp()
398 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {1}, quant); in SetUp()
399 interpreter_->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", {1}, quant); in SetUp()
404 interpreter_->SetTensorParametersReadWrite(5, kTfLiteFloat32, "", {1}, quant); in SetUp()
405 interpreter_->SetTensorParametersReadWrite(6, kTfLiteFloat32, "", {1}, quant); in SetUp()
[all …]
Ddelegate_test.cc152 ASSERT_NE(interpreter_->SetTensorParametersReadWrite(0, kTfLiteInt8, "", {3}, in TEST_F()
862 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {3}, in SetUp()
864 interpreter_->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", {3}, in SetUp()
866 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {3}, in SetUp()
1007 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {3}, in SetUp()
1009 interpreter_->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", {3}, in SetUp()
1011 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {3}, in SetUp()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc36 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {3}, in SetUp()
38 interpreter_->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", {3}, in SetUp()
40 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {3}, in SetUp()
42 interpreter_->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", {3}, in SetUp()
44 interpreter_->SetTensorParametersReadWrite(4, kTfLiteFloat32, "", {3}, in SetUp()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h48 interpreter->SetTensorParametersReadWrite( in resize()
51 interpreter->SetTensorParametersReadWrite(1, kTfLiteInt32, "new_size", {2}, in resize()
53 interpreter->SetTensorParametersReadWrite( in resize()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc256 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
260 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
273 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
277 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
519 EXPECT_EQ(interpreter_.SetTensorParametersReadWrite( in InterpreterFp32()
523 interpreter_.SetTensorParametersReadWrite( in InterpreterFp32()
527 interpreter_.SetTensorParametersReadWrite( in InterpreterFp32()
665 EXPECT_EQ(interpreter_.SetTensorParametersReadWrite( in Interpreter2Fp32()
669 interpreter_.SetTensorParametersReadWrite( in Interpreter2Fp32()
673 interpreter_.SetTensorParametersReadWrite( in Interpreter2Fp32()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h128 inline TfLiteStatus SetTensorParametersReadWrite(
133 return SetTensorParametersReadWrite(tensor_index, type, name, dims.size(),
137 return SetTensorParametersReadWrite(
141 TfLiteStatus SetTensorParametersReadWrite(
Dsubgraph.cc1507 TfLiteStatus Subgraph::SetTensorParametersReadWrite( in SetTensorParametersReadWrite() function in tflite::Subgraph
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc80 CHECK_EQ(interpreter_->SetTensorParametersReadWrite(i, type, in AddTensors()
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall_test.cc42 ASSERT_EQ(subgraph->SetTensorParametersReadWrite(tensor_index, type, "", 0, in SetupTensor()