Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc62 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
87 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
106 AddTensors(3, {0, 1}, {2}, kTfLiteString, {2}); in TEST_F()
125 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
147 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
171 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
189 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
218 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
236 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
266 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
[all …]
Dkernel_test.cc79 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
130 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
174 AddTensors(10, {0, 3}, {9}, kTfLiteFloat32, {3}); in TEST_F()
200 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
211 AddTensors(3, {0}, {1, 2}, kTfLiteFloat32, {3}); in TEST_F()
225 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
238 AddTensors(4, {0}, {3}, kTfLiteFloat32, {3}); in TEST_F()
253 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
282 AddTensors(18, {0, 1}, {17}, kTfLiteFloat32, {3}); in TEST_F()
358 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
[all …]
Dtest_util.cc74 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors() function in tflite::flex::testing::FlexModelTest
77 interpreter_->AddTensors(num_tensors); in AddTensors()
Dtest_util.h94 void AddTensors(int num_tensors, const std::vector<int>& inputs,
/external/tensorflow/tensorflow/lite/
Dgraph_info_test.cc79 void AddTensors(int count) { tensors_.resize(count + tensors_.size()); } in AddTensors() function in tflite::__anon1abf07070111::SimpleTestGraph
154 graph.AddTensors(1); in TEST()
167 graph.AddTensors(2); in TEST()
185 graph.AddTensors(2); in TEST()
205 graph.AddTensors(1); in TEST()
225 graph.AddTensors(2); in TEST()
247 graph.AddTensors(3); in TEST()
274 graph.AddTensors(3); in TEST()
302 graph.AddTensors(3); in TEST()
332 graph.AddTensors(4); in TEST()
[all …]
Dinterpreter_test.cc96 ASSERT_EQ(interpreter.AddTensors(1, &tensor_index), kTfLiteOk); in TEST()
114 ASSERT_EQ(interpreter.AddTensors(1, &tensor_index), kTfLiteOk); in TEST()
137 ASSERT_EQ(interpreter.AddTensors(2, &base_index), kTfLiteOk); in TEST()
140 ASSERT_EQ(interpreter.AddTensors(3, &base_index), kTfLiteOk); in TEST()
142 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
179 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
207 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
224 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
296 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
332 ASSERT_EQ(interpreter.AddTensors(4), kTfLiteOk); in TEST()
[all …]
Dstring_util_test.cc30 interpreter.AddTensors(3); in TEST()
106 interpreter.AddTensors(1); in TEST()
130 interpreter.AddTensors(1); in TEST()
155 interpreter.AddTensors(1); in TEST()
168 interpreter.AddTensors(1); in TEST()
Dinterpreter.cc244 TfLiteStatus Interpreter::AddTensors(int tensors_to_add, in AddTensors() function in tflite::Interpreter
246 return primary_subgraph().AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
/external/tensorflow/tensorflow/lite/delegates/flex/training/
Dtraining_delegate_test.cc65 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
84 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
108 AddTensors(2, {0}, {1}, kTfLiteBool, {1}); in TEST_F()
127 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/shim/
Dtflite_tensor_view_test.cc35 interpreter.AddTensors(1); in TEST()
57 interpreter.AddTensors(1); in IntTest()
85 interpreter.AddTensors(1); in FloatTest()
108 interpreter.AddTensors(1); in TEST()
151 interpreter.AddTensors(1); in TEST()
Dtest_util_test.cc28 interpreter.AddTensors(3); in TEST()
46 interpreter.AddTensors(2); in TEST()
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc100 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAddSubgraph()
131 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMulSubgraph()
162 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadSubgraph()
193 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildIfSubgraph()
230 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildLessEqualCondSubgraph()
267 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAccumulateLoopBodySubgraph()
314 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadLoopBodySubgraph()
361 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildWhileSubgraph()
395 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAssignRandomValueToVariableSubgraph()
422 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildCallOnceAndReadVariableSubgraph()
[all …]
Dvariable_ops_test.cc57 ASSERT_EQ(interpreter_->AddTensors(3, &first_new_tensor_index), kTfLiteOk); in ConstructInvalidGraph()
92 ASSERT_EQ(interpreter_->AddTensors(3, &first_new_tensor_index), kTfLiteOk); in ConstructGraph()
123 ASSERT_EQ(interpreter_->AddTensors(3, &first_new_tensor_index), kTfLiteOk); in ConstructGraphWithKnownShape()
Dhashtable_ops_test.cc326 void AddTensors(bool table_two_initialization = false) { in AddTensors() function in tflite::__anon44220aca0111::HashtableGraph
329 ASSERT_EQ(interpreter_->AddTensors(7, &first_new_tensor_index), in AddTensors()
338 ASSERT_EQ(interpreter_->AddTensors(14, &first_new_tensor_index), in AddTensors()
515 graph_->AddTensors(); in HashtableDefaultGraphTest()
576 graph.AddTensors(); in TEST()
589 graph.AddTensors(); in TEST()
605 graph.AddTensors(/*table_two_initialization=*/true); in TEST()
621 graph.AddTensors(); in TEST()
Dconv.cc273 context->AddTensors(context, 1, &data->im2col_id); in AllocateTemporaryTensorsIfRequired()
280 context->AddTensors(context, 1, &data->hwcn_weights_id); in AllocateTemporaryTensorsIfRequired()
290 context, context->AddTensors(context, 1, &data->input_quantized_id)); in AllocateTemporaryTensorsIfRequired()
299 context, context->AddTensors(context, 1, &data->scaling_factors_id)); in AllocateTemporaryTensorsIfRequired()
307 context, context->AddTensors(context, 1, &data->accum_scratch_id)); in AllocateTemporaryTensorsIfRequired()
314 context, context->AddTensors(context, 1, &data->input_offset_id)); in AllocateTemporaryTensorsIfRequired()
321 context->AddTensors(context, 1, &data->row_sums_id)); in AllocateTemporaryTensorsIfRequired()
Dconv3d.cc102 context, context->AddTensors(context, 1, &opdata->im2col_tensor_id)); in AllocateTemporaryTensorsIfRequired()
110 context, context->AddTensors(context, 1, in AllocateTemporaryTensorsIfRequired()
Dadd_n.cc41 context->AddTensors(context, 1, &op_data->scratch_tensor_index); in Init()
Ddepthwise_conv.cc215 context, context->AddTensors(context, 1, &data->input_quantized_id)); in Prepare()
221 context, context->AddTensors(context, 1, &data->scaling_factors_id)); in Prepare()
227 context, context->AddTensors(context, 1, &data->input_offset_id)); in Prepare()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph_test.cc38 subgraph.AddTensors(4); in TEST()
53 subgraph.AddTensors(4); in TEST()
66 subgraph.AddTensors(3); in TEST()
Dsubgraph.h102 TfLiteStatus AddTensors(int tensors_to_add,
571 static TfLiteStatus AddTensors(TfLiteContext* context, int tensors_to_add,
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc67 interpreter.AddTensors(3); in TEST_P()
110 interpreter.AddTensors(3); in TEST_P()
144 interpreter.AddTensors(4); in TEST_P()
191 interpreter.AddTensors(5); in TEST_P()
246 interpreter.AddTensors(3); in TEST_P()
286 interpreter.AddTensors(3); in TEST_P()
317 interpreter.AddTensors(3); in TEST_P()
354 interpreter.AddTensors(3); in TEST_P()
411 interpreter.AddTensors(total_tensors); in TEST_P()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h39 interpreter->AddTensors(2, &base_index); in resize()
41 interpreter->AddTensors(1, &base_index); in resize()
/external/armnn/delegate/classic/src/test/
DArmnnClassicDelegateTest.cpp26 tfLiteInterpreter->AddTensors(3);
64 tfLiteInterpreter->AddTensors(3);
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall_test.cc51 ASSERT_EQ(subgraph->AddTensors(inputs.size() + outputs.size(), in BuildCallSubgraph()
101 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildGraphWithMultipleOutputs()
141 ASSERT_EQ(this_subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMultiNodeGraph()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc32 interpreter_->AddTensors(5); in SetUp()

123