Home
last modified time | relevance | path

Searched refs:first_new_tensor_index (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc104 int first_new_tensor_index; in BuildAddSubgraph() local
105 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAddSubgraph()
107 ASSERT_EQ(first_new_tensor_index, 0); in BuildAddSubgraph()
135 int first_new_tensor_index; in BuildMulSubgraph() local
136 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMulSubgraph()
138 ASSERT_EQ(first_new_tensor_index, 0); in BuildMulSubgraph()
166 int first_new_tensor_index; in BuildPadSubgraph() local
167 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadSubgraph()
169 ASSERT_EQ(first_new_tensor_index, 0); in BuildPadSubgraph()
197 int first_new_tensor_index; in BuildIfSubgraph() local
[all …]
Dvariable_ops_test.cc53 int first_new_tensor_index; in ConstructGraph() local
54 ASSERT_EQ(interpreter_.AddTensors(4, &first_new_tensor_index), kTfLiteOk); in ConstructGraph()
/external/tensorflow/tensorflow/lite/delegates/
Dutils_test.cc38 int* first_new_tensor_index) { in TEST() argument
45 *first_new_tensor_index = 1; in TEST()
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc299 int* first_new_tensor_index) { in AddTensors() argument
300 return primary_subgraph().AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
Dinterpreter.h144 int* first_new_tensor_index = nullptr);
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h96 int* first_new_tensor_index = nullptr);
487 int* first_new_tensor_index);
Dsubgraph.cc1152 int* first_new_tensor_index) { in AddTensors() argument
1154 if (first_new_tensor_index) *first_new_tensor_index = base_index; in AddTensors()
1166 int* first_new_tensor_index) { in AddTensors() argument
1171 ->AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_ops_test.cc332 int first_new_tensor_index; in AddTensors() local
334 ASSERT_EQ(interpreter_->AddTensors(7, &first_new_tensor_index), in AddTensors()
343 ASSERT_EQ(interpreter_->AddTensors(14, &first_new_tensor_index), in AddTensors()
/external/tensorflow/tensorflow/lite/c/
Dcommon.h660 int* first_new_tensor_index);