Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util_test.cc48 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {input1}); in TestAccumelateLoopBody()
49 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {input2}); in TestAccumelateLoopBody()
71 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
72 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
86 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
87 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
101 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {5, 7}); in TEST_F()
102 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1, 2}); in TEST_F()
118 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), in TEST_F()
138 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
[all …]
Dwhile_test.cc28 using subgraph_test_util::FillIntTensor;
51 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
52 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {1}); in TEST_F()
77 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[0]), {1}); in TEST_F()
78 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in TEST_F()
Dif_test.cc29 using subgraph_test_util::FillIntTensor;
48 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in SetUp()
49 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {1, 2}); in SetUp()
82 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), {5, 7}); in SetUp()
83 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[2]), {1, 2}); in SetUp()
Dsubgraph_test_util.h111 void FillIntTensor(TfLiteTensor* tensor, const std::vector<int32_t>& data);
Dsubgraph_test_util.cc372 void FillIntTensor(TfLiteTensor* tensor, const std::vector<int32_t>& data) { in FillIntTensor() function