Home
last modified time | relevance | path

Searched refs:SetInputs (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dnumeric_verify_test.cc71 void SetInputs(std::initializer_list<T> data, in SetInputs() function in tflite::__anondc71b06f0111::NumericVerifyOpModel
89 m.SetInputs<uint8_t>({0, 1, 2, 3, 4, 251, 252, 253, 254, 255}, in TEST()
98 m.SetInputs<int8_t>({-128, -127, -126, -125, -124, 123, 124, 125, 126, 127}, in TEST()
118 m.SetInputs<int16_t>( in TEST()
129 m.SetInputs<int8_t>({-128, -127, -126, -125, -124, 0, 124, 125, 126, 127}, in TEST()
139 m.SetInputs<int8_t>({-128, -127, -126, -125, -124, 0, 124, 125, 126, 127}, in TEST()
Dsubgraph_test_util.cc108 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildAddSubgraph()
139 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildMulSubgraph()
170 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildPadSubgraph()
201 ASSERT_EQ(subgraph->SetInputs({kCondInput, kInput1, kInput2}), kTfLiteOk); in BuildIfSubgraph()
238 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildLessEqualCondSubgraph()
275 ASSERT_EQ(subgraph->SetInputs({kInputCounter, kInputValue}), kTfLiteOk); in BuildAccumulateLoopBodySubgraph()
322 ASSERT_EQ(subgraph->SetInputs({kInputCounter, kInputValue}), kTfLiteOk); in BuildPadLoopBodySubgraph()
369 ASSERT_EQ(subgraph->SetInputs({kInput1, kInput2}), kTfLiteOk); in BuildWhileSubgraph()
402 ASSERT_EQ(subgraph->SetInputs({}), kTfLiteOk); in BuildAssignRandomValueToVariableSubgraph()
429 ASSERT_EQ(subgraph->SetInputs({}), kTfLiteOk); in BuildCallOnceAndReadVariableSubgraph()
[all …]
Dvariable_ops_test.cc55 ASSERT_EQ(interpreter_.SetInputs({0, 1, 2}), kTfLiteOk); in ConstructGraph()
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc71 interpreter.SetInputs({}); in TEST()
150 ASSERT_NE(interpreter.SetInputs({5}), kTfLiteOk); in TEST()
178 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
201 interpreter.SetInputs({0, 1}); in TEST()
218 interpreter.SetInputs({0, 1}); in TEST()
291 interpreter.SetInputs({0, 1}); in TEST()
354 interpreter.SetInputs({0, 1}); in TEST()
378 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
395 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
411 ASSERT_EQ(interpreter.SetInputs({0}), kTfLiteOk); in TEST()
[all …]
Dinterpreter.cc169 TfLiteStatus Interpreter::SetInputs(std::vector<int> inputs) { in SetInputs() function in tflite::Interpreter
170 return primary_subgraph().SetInputs(std::move(inputs)); in SetInputs()
Dinterpreter.h112 TfLiteStatus SetInputs(std::vector<int> inputs);
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc70 interpreter.SetInputs({0, 1}); in TEST_P()
113 interpreter.SetInputs({0, 1}); in TEST_P()
149 interpreter.SetInputs({0, 1}); in TEST_P()
198 interpreter.SetInputs({0, 1}); in TEST_P()
248 interpreter.SetInputs({0}); in TEST_P()
287 interpreter.SetInputs({0, 1}); in TEST_P()
351 interpreter.SetInputs(input_tensors); in TEST_P()
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc82 void SetInputs(int32_t x, int32_t y) { in SetInputs() function in tflite::profiling::__anonfc60a69d0111::SimpleOpModel
115 m.SetInputs(1, 2); in TEST()
136 m.SetInputs(1, 2); in TEST()
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dobject_detection_stage.h63 void SetInputs(const std::string& image_path, in SetInputs() function
Dinference_profiler_stage.cc161 test_stage_->SetInputs(input_ptrs); in Run()
162 reference_stage_->SetInputs(input_ptrs); in Run()
Dtflite_inference_stage.h57 void SetInputs(const std::vector<void*>& raw_input_ptrs) { in SetInputs() function
Dimage_classification_stage.h57 void SetInputs(const std::string& image_path, in SetInputs() function
Dobject_detection_stage.cc104 inference_stage_->SetInputs(data_ptrs); in Run()
Dimage_classification_stage.cc115 inference_stage_->SetInputs(data_ptrs); in Run()
Dtflite_inference_stage_test.cc171 stage.SetInputs(inputs); in TEST()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h43 interpreter->SetInputs({0, 1}); in resize()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc195 EXPECT_EQ(interpreter_.SetInputs({0, 1}), kTfLiteOk); in InterpreterFp16()
468 EXPECT_EQ(interpreter_.SetInputs({0, 2}), kTfLiteOk); in InterpreterFp32()
583 EXPECT_EQ(interpreter_.SetInputs({0, 2, 4, 6}), kTfLiteOk); in Interpreter2Fp32()
763 EXPECT_EQ(interpreter_.SetInputs({0, 1, 2}), kTfLiteOk); in InterpreterMultiNode()
1083 EXPECT_EQ(interpreter_.SetInputs({0, 3}), kTfLiteOk); in InterpreterQuantized()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc32 interpreter_->SetInputs({0, 1}); in SetUp()
/external/tensorflow/tensorflow/lite/tools/evaluation/tasks/imagenet_image_classification/
Drun_eval.cc168 eval.SetInputs(image_labels[i].image, image_labels[i].label); in RunImpl()
/external/tensorflow/tensorflow/lite/tools/evaluation/tasks/coco_object_detection/
Drun_eval.cc152 eval.SetInputs(image_paths[i], ground_truth_map[image_name]); in RunImpl()
/external/tensorflow/tensorflow/c/
Dc_test_util.h140 void SetInputs(std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs);
Dwhile_loop_test.cc89 csession_->SetInputs(inputs); in Run()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc99 subgraph->SetInputs({0, 1}); in SetUpSubgraph()
334 interpreter_->SetInputs({0}); in SetUp()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc91 CHECK_EQ(interpreter_->SetInputs(inputs), kTfLiteOk); in AddTensors()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparallel_interleave_dataset_op.cc362 current_worker->SetInputs(s, std::move(args)); in GetNextInternal()
611 void SetInputs(const Status& s, std::vector<Tensor> input_arguments) { in SetInputs() function
669 workers_[i].SetInputs(s, std::move(args)); in EnsureWorkerThreadsStarted()

12