Searched refs:FillIntTensor (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | subgraph_test_util_test.cc | 55 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 …]
|
H A D | if_test.cc | 30 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() 77 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), input_vector); in TEST_F() 78 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {9}); in TEST_F() 86 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {19}); in TEST_F() 103 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), input_vector); in TEST_F() 104 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {0}); in TEST_F() 112 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {7}); in TEST_F() 134 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in SetUp() [all …]
|
H A D | while_test.cc | 31 using subgraph_test_util::FillIntTensor; 52 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F() 53 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1}); in TEST_F() 92 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F() 94 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), in TEST_F() 124 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F() 125 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in TEST_F() 151 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F() 155 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), input_vector); in TEST_F() 188 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {1}); in TEST_F()
|
H A D | subgraph_test_util.h | 144 void FillIntTensor(TfLiteTensor* tensor, const std::vector<int32_t>& data);
|
H A D | subgraph_test_util.cc | 641 void FillIntTensor(TfLiteTensor* tensor, const std::vector<int32_t>& data) { in FillIntTensor() function
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
H A D | call_test.cc | 200 subgraph_test_util::FillIntTensor( in TEST_F() 202 subgraph_test_util::FillIntTensor( in TEST_F() 246 subgraph_test_util::FillIntTensor( in TEST_F() 248 subgraph_test_util::FillIntTensor( in TEST_F() 271 subgraph_test_util::FillIntTensor( in TEST_F() 273 subgraph_test_util::FillIntTensor( in TEST_F() 318 subgraph_test_util::FillIntTensor( in TEST_F() 320 subgraph_test_util::FillIntTensor( in TEST_F() 322 subgraph_test_util::FillIntTensor( in TEST_F() 325 subgraph_test_util::FillIntTensor( in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/ |
H A D | writer_lib_test.cc | 42 using subgraph_test_util::FillIntTensor; 532 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F() 564 FillIntTensor(new_interpreter->tensor(interpreter_->inputs()[0]), {1}); in TEST_F() 565 FillIntTensor(new_interpreter->tensor(interpreter_->inputs()[1]), {1}); in TEST_F() 588 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/profiling/ |
H A D | profile_summarizer_test.cc | 167 subgraph_test_util::FillIntTensor( in SetUp() 169 subgraph_test_util::FillIntTensor( in SetUp()
|