Searched refs:first_new_tensor_index (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | subgraph_test_util.cc | 99 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 D | variable_ops_test.cc | 56 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 D | hashtable_ops_test.cc | 327 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 D | call_test.cc | 50 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 D | utils_test.cc | 39 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 D | interpreter.cc | 245 int* first_new_tensor_index) { in AddTensors() argument 246 return primary_subgraph().AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
|
H A D | interpreter.h | 151 int* first_new_tensor_index = nullptr);
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/ |
H A D | subgraph.cc | 1409 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 D | subgraph.h | 103 int* first_new_tensor_index = nullptr); 572 int* first_new_tensor_index);
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
H A D | common.h | 723 int* first_new_tensor_index);
|
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
H A D | common.h | 771 int* first_new_tensor_index);
|