Home
last modified time | relevance | path

Searched refs:AddTensors (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc52 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
77 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
96 AddTensors(3, {0, 1}, {2}, kTfLiteString, {2}); in TEST_F()
115 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
137 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
161 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
179 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
208 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
226 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
256 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
[all …]
Dkernel_test.cc103 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
140 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
153 AddTensors(3, {0}, {1, 2}, kTfLiteFloat32, {3}); in TEST_F()
169 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
187 AddTensors(4, {0}, {3}, kTfLiteFloat32, {3}); in TEST_F()
202 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
233 AddTensors(18, {0, 1}, {17}, kTfLiteFloat32, {3}); in TEST_F()
317 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
350 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
386 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
Dtest_util.cc70 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors() function in tflite::flex::testing::FlexModelTest
73 interpreter_->AddTensors(num_tensors); in AddTensors()
Dtest_util.h88 void AddTensors(int num_tensors, const std::vector<int>& inputs,
/external/tensorflow/tensorflow/lite/
Dgraph_info_test.cc58 void AddTensors(int count) { tensors_.resize(count + tensors_.size()); } in AddTensors() function in tflite::__anon531b66cb0111::SimpleTestGraph
118 graph.AddTensors(1); in TEST()
131 graph.AddTensors(2); in TEST()
151 graph.AddTensors(1); in TEST()
171 graph.AddTensors(2); in TEST()
193 graph.AddTensors(3); in TEST()
221 graph.AddTensors(3); in TEST()
251 graph.AddTensors(4); in TEST()
Dinterpreter_test.cc87 ASSERT_EQ(interpreter.AddTensors(1, &tensor_index), kTfLiteOk); in TEST()
107 ASSERT_EQ(interpreter.AddTensors(2, &base_index), kTfLiteOk); in TEST()
110 ASSERT_EQ(interpreter.AddTensors(3, &base_index), kTfLiteOk); in TEST()
112 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
149 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
172 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
189 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
256 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
293 ASSERT_EQ(interpreter.AddTensors(4), kTfLiteOk); in TEST()
310 ASSERT_EQ(interpreter.AddTensors(10), kTfLiteOk); in TEST()
[all …]
Dstring_util_test.cc26 interpreter.AddTensors(3); in TEST()
93 interpreter.AddTensors(1); in TEST()
116 interpreter.AddTensors(1); in TEST()
129 interpreter.AddTensors(1); in TEST()
Dinterpreter.cc146 TfLiteStatus Interpreter::AddTensors(int tensors_to_add, in AddTensors() function in tflite::Interpreter
148 return primary_subgraph().AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
Dinterpreter.h156 TfLiteStatus AddTensors(int tensors_to_add,
Dmodel.cc537 if (modified_subgraph->AddTensors(tensors->Length()) != kTfLiteOk) { in operator ()()
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc70 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAddSubgraph()
100 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMulSubgraph()
130 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadSubgraph()
160 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildIfSubgraph()
200 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildLessEqualCondSubgraph()
235 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAccumulateLoopBodySubgraph()
280 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadLoopBodySubgraph()
323 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildWhileSubgraph()
Dconv.cc178 context->AddTensors(context, 1, &data->im2col_id); in AllocateTemporaryTensorsIfRequired()
185 context->AddTensors(context, 1, &data->hwcn_weights_id); in AllocateTemporaryTensorsIfRequired()
195 context, context->AddTensors(context, 1, &data->input_quantized_id)); in AllocateTemporaryTensorsIfRequired()
204 context, context->AddTensors(context, 1, &data->scaling_factors_id)); in AllocateTemporaryTensorsIfRequired()
Ddetection_postprocess.cc118 context->AddTensors(context, 1, &op_data->decoded_boxes_index); in Init()
119 context->AddTensors(context, 1, &op_data->scores_index); in Init()
120 context->AddTensors(context, 1, &op_data->active_candidate_index); in Init()
Dtranspose_conv.cc95 context->AddTensors(context, 1, &data->im2col_id); in AllocateIm2colTensorIfRequired()
Dbasic_rnn.cc51 context->AddTensors(context, /*tensors_to_add=*/3, scratch_tensor_index); in Init()
Dunidirectional_sequence_rnn.cc46 context->AddTensors(context, /*tensors_to_add=*/3, scratch_tensor_index); in Init()
Dsvdf.cc116 context->AddTensors(context, /*tensors_to_add=*/4, in Init()
Dfully_connected.cc73 context->AddTensors(context, /*tensors_to_add=*/2, in Init()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h40 interpreter->AddTensors(2, &base_index); in resize()
42 interpreter->AddTensors(1, &base_index); in resize()
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib_test.cc28 interpreter.AddTensors(3); in TEST()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc150 context_->AddTensors = AddTensors; in Subgraph()
777 TfLiteStatus Subgraph::AddTensors(int tensors_to_add, in AddTensors() function in tflite::Subgraph
791 TfLiteStatus Subgraph::AddTensors(TfLiteContext* context, int tensors_to_add, in AddTensors() function in tflite::Subgraph
797 ->AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
Dsubgraph.h81 TfLiteStatus AddTensors(int tensors_to_add,
376 static TfLiteStatus AddTensors(TfLiteContext* context, int tensors_to_add,
/external/tensorflow/tensorflow/lite/experimental/micro/testing/
Dtest_utils.h89 context->AddTensors = nullptr; in PopulateContext()
/external/tensorflow/tensorflow/lite/c/
Dc_api_internal.h431 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, member
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter.cc150 context_.AddTensors = nullptr; in MicroInterpreter()

12