Home
last modified time | relevance | path

Searched refs:FillIntTensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util_test.cc55 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {input1}); in TestAccumulateLoopBody()
56 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {input2}); in TestAccumulateLoopBody()
78 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
79 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
93 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
94 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
108 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
109 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
123 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
124 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
[all …]
Dwhile_test.cc30 using subgraph_test_util::FillIntTensor;
54 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
55 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1}); in TEST_F()
76 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
77 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in TEST_F()
107 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {1}); in TEST_F()
Dif_test.cc30 using subgraph_test_util::FillIntTensor;
49 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in SetUp()
50 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {1, 2}); in SetUp()
83 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in SetUp()
84 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {1, 2}); in SetUp()
Dsubgraph_test_util.h144 void FillIntTensor(TfLiteTensor* tensor, const std::vector<int32_t>& data);
Dsubgraph_test_util.cc602 void FillIntTensor(TfLiteTensor* tensor, const std::vector<int32_t>& data) { in FillIntTensor() function
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc37 using subgraph_test_util::FillIntTensor;
444 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
476 FillIntTensor(new_interpreter->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
477 FillIntTensor(new_interpreter->tensor(interpreter_->inputs()[1]), {1}); in TEST_F()
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc166 subgraph_test_util::FillIntTensor( in SetUp()
168 subgraph_test_util::FillIntTensor( in SetUp()