Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc63 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
68 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
106 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
111 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
140 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
145 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
147 interpreter.SetTensorParametersReadWrite(3, kTfLiteFloat32, "d", {3}, in TEST_P()
187 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
192 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
194 interpreter.SetTensorParametersReadWrite(3, kTfLiteFloat32, "d", {3}, in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc241 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
245 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
258 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
262 interpreter_.SetTensorParametersReadWrite( in InterpreterFp16()
506 EXPECT_EQ(interpreter_.SetTensorParametersReadWrite( in InterpreterFp32()
510 interpreter_.SetTensorParametersReadWrite( in InterpreterFp32()
514 interpreter_.SetTensorParametersReadWrite( in InterpreterFp32()
652 EXPECT_EQ(interpreter_.SetTensorParametersReadWrite( in Interpreter2Fp32()
656 interpreter_.SetTensorParametersReadWrite( in Interpreter2Fp32()
660 interpreter_.SetTensorParametersReadWrite( in Interpreter2Fp32()
[all …]
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc96 interpreter.SetTensorParametersReadWrite(tensor_index, kTfLiteFloat32, "", in TEST()
116 interpreter.SetTensorParametersReadWrite(tensor_index, kTfLiteInt8, "", in TEST()
205 interpreter.SetTensorParametersReadWrite(0, test.type, "", {3}, quant); in TEST()
206 interpreter.SetTensorParametersReadWrite(1, test.type, "", {4}, quant); in TEST()
245 ASSERT_EQ(interpreter.SetTensorParametersReadWrite(0, tensor_type, "", {3}, in TEST()
296 interpreter.SetTensorParametersReadWrite(0, test.type, "", {3}, quant), in TEST()
330 interpreter.SetTensorParametersReadWrite(i, test.type, "", {2 * i + 1}, in TEST()
351 interpreter.SetTensorParametersReadWrite(static_cast<int>(i), kTfLiteUInt8, in TEST()
380 ASSERT_EQ(interpreter.SetTensorParametersReadWrite( in TEST()
398 ASSERT_EQ(interpreter.SetTensorParametersReadWrite( in TEST()
[all …]
Dinterpreter.cc316 TfLiteStatus Interpreter::SetTensorParametersReadWrite( in SetTensorParametersReadWrite() function in tflite::Interpreter
320 return primary_subgraph().SetTensorParametersReadWrite( in SetTensorParametersReadWrite()
335 TfLiteStatus Interpreter::SetTensorParametersReadWrite( in SetTensorParametersReadWrite() function in tflite::Interpreter
340 return primary_subgraph().SetTensorParametersReadWrite( in SetTensorParametersReadWrite()
Dinterpreter.h175 TfLiteStatus SetTensorParametersReadWrite(int tensor_index, TfLiteType type,
182 inline TfLiteStatus SetTensorParametersReadWrite(
193 return SetTensorParametersReadWrite(
197 TfLiteStatus SetTensorParametersReadWrite(
Dinterpreter_builder.cc629 if (subgraph->SetTensorParametersReadWrite( in ParseTensors()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc103 subgraph->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
105 subgraph->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
107 subgraph->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
109 subgraph->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
111 subgraph->SetTensorParametersReadWrite(4, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
342 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {1}, quant); in SetUp()
347 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {1}, quant); in SetUp()
348 interpreter_->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", {1}, quant); in SetUp()
353 interpreter_->SetTensorParametersReadWrite(5, kTfLiteFloat32, "", {1}, quant); in SetUp()
354 interpreter_->SetTensorParametersReadWrite(6, kTfLiteFloat32, "", {1}, quant); in SetUp()
[all …]
Ddelegate_test.cc252 ASSERT_NE(interpreter_->SetTensorParametersReadWrite(0, kTfLiteInt8, "", {3}, in TEST_F()
816 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {3}, in SetUp()
818 interpreter_->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", {3}, in SetUp()
820 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {3}, in SetUp()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc35 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {3}, in SetUp()
37 interpreter_->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", {3}, in SetUp()
39 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {3}, in SetUp()
41 interpreter_->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", {3}, in SetUp()
43 interpreter_->SetTensorParametersReadWrite(4, kTfLiteFloat32, "", {3}, in SetUp()
/external/tensorflow/tensorflow/lite/kernels/
Dvariable_ops_test.cc57 interpreter_.SetTensorParametersReadWrite(0, kTfLiteInt32, "", 0, nullptr, in ConstructGraph()
59 interpreter_.SetTensorParametersReadWrite(1, kTfLiteInt32, "", 0, nullptr, in ConstructGraph()
61 interpreter_.SetTensorParametersReadWrite(2, kTfLiteFloat32, "", 0, nullptr, in ConstructGraph()
63 interpreter_.SetTensorParametersReadWrite(3, kTfLiteFloat32, "", 0, nullptr, in ConstructGraph()
Dsubgraph_test_util.cc82 ASSERT_EQ(subgraph->SetTensorParametersReadWrite(tensor_index, type, "", 0, in SetupTensor()
/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/kernels/hashtable/
Dhashtable_ops_test.cc358 interpreter_->SetTensorParametersReadWrite( in AddTensors()
362 interpreter_->SetTensorParametersReadWrite(kKeyTensorId, key_type_, "", in AddTensors()
367 interpreter_->SetTensorParametersReadWrite( in AddTensors()
372 interpreter_->SetTensorParametersReadWrite( in AddTensors()
377 interpreter_->SetTensorParametersReadWrite( in AddTensors()
382 interpreter_->SetTensorParametersReadWrite(kSizeTensorId, kTfLiteInt64, "", in AddTensors()
386 interpreter_->SetTensorParametersReadWrite( in AddTensors()
391 interpreter_->SetTensorParametersReadWrite( in AddTensors()
395 interpreter_->SetTensorParametersReadWrite( in AddTensors()
400 interpreter_->SetTensorParametersReadWrite( in AddTensors()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h121 inline TfLiteStatus SetTensorParametersReadWrite(
126 return SetTensorParametersReadWrite(tensor_index, type, name, dims.size(),
130 return SetTensorParametersReadWrite(
134 TfLiteStatus SetTensorParametersReadWrite(
Dsubgraph.cc1252 TfLiteStatus Subgraph::SetTensorParametersReadWrite( in SetTensorParametersReadWrite() function in tflite::Subgraph
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc85 CHECK_EQ(interpreter_->SetTensorParametersReadWrite(i, type, in AddTensors()