Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsubgraph_test_util.cc99 int first_new_tensor_index; in BuildAddSubgraph() local
100 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAddSubgraph()
102 ASSERT_EQ(first_new_tensor_index, 0); in BuildAddSubgraph()
130 int first_new_tensor_index; in BuildMulSubgraph() local
131 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMulSubgraph()
133 ASSERT_EQ(first_new_tensor_index, 0); in BuildMulSubgraph()
161 int first_new_tensor_index; in BuildPadSubgraph() local
162 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadSubgraph()
164 ASSERT_EQ(first_new_tensor_index, 0); in BuildPadSubgraph()
192 int first_new_tensor_index; in BuildIfSubgraph() local
[all …]
H A Dvariable_ops_test.cc56 int first_new_tensor_index; in ConstructInvalidGraph() local
57 ASSERT_EQ(interpreter_->AddTensors(3, &first_new_tensor_index), kTfLiteOk); in ConstructInvalidGraph()
91 int first_new_tensor_index; in ConstructGraph() local
92 ASSERT_EQ(interpreter_->AddTensors(3, &first_new_tensor_index), kTfLiteOk); in ConstructGraph()
122 int first_new_tensor_index; in ConstructGraphWithKnownShape() local
123 ASSERT_EQ(interpreter_->AddTensors(3, &first_new_tensor_index), kTfLiteOk); in ConstructGraphWithKnownShape()
H A Dhashtable_ops_test.cc327 int first_new_tensor_index; in AddTensors() local
329 ASSERT_EQ(interpreter_->AddTensors(7, &first_new_tensor_index), in AddTensors()
338 ASSERT_EQ(interpreter_->AddTensors(14, &first_new_tensor_index), in AddTensors()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall_test.cc50 int first_new_tensor_index; in BuildCallSubgraph() local
52 &first_new_tensor_index), in BuildCallSubgraph()
54 ASSERT_EQ(first_new_tensor_index, 0); in BuildCallSubgraph()
100 int first_new_tensor_index; in BuildGraphWithMultipleOutputs() local
101 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildGraphWithMultipleOutputs()
103 ASSERT_EQ(first_new_tensor_index, 0); in BuildGraphWithMultipleOutputs()
140 int first_new_tensor_index; in BuildMultiNodeGraph() local
141 ASSERT_EQ(this_subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMultiNodeGraph()
143 ASSERT_EQ(first_new_tensor_index, 0); in BuildMultiNodeGraph()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils_test.cc39 int* first_new_tensor_index) { in TEST() argument
46 *first_new_tensor_index = 1; in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter.cc245 int* first_new_tensor_index) { in AddTensors() argument
246 return primary_subgraph().AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
H A Dinterpreter.h151 int* first_new_tensor_index = nullptr);
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc1409 int* first_new_tensor_index) { in AddTensors() argument
1411 if (first_new_tensor_index) *first_new_tensor_index = base_index; in AddTensors()
1423 int* first_new_tensor_index) { in AddTensors() argument
1428 ->AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
H A Dsubgraph.h103 int* first_new_tensor_index = nullptr);
572 int* first_new_tensor_index);
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon.h723 int* first_new_tensor_index);
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/
H A Dcommon.h771 int* first_new_tensor_index);