Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc55 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
80 AddTensors(3, {0, 1}, {2}, kTfLiteInt32, {2}); in TEST_F()
99 AddTensors(3, {0, 1}, {2}, kTfLiteString, {2}); in TEST_F()
118 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
140 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
164 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
182 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
211 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
229 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
259 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
[all …]
Dkernel_test.cc62 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
96 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
107 AddTensors(3, {0}, {1, 2}, kTfLiteFloat32, {3}); in TEST_F()
121 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
137 AddTensors(4, {0}, {3}, kTfLiteFloat32, {3}); in TEST_F()
152 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
181 AddTensors(18, {0, 1}, {17}, kTfLiteFloat32, {3}); in TEST_F()
257 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
290 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
326 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
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.h93 void AddTensors(int num_tensors, const std::vector<int>& inputs,
/external/tensorflow/tensorflow/lite/
Dgraph_info_test.cc77 void AddTensors(int count) { tensors_.resize(count + tensors_.size()); } in AddTensors() function in tflite::__anonb6ed36e90111::SimpleTestGraph
138 graph.AddTensors(1); in TEST()
151 graph.AddTensors(2); in TEST()
169 graph.AddTensors(2); in TEST()
189 graph.AddTensors(1); in TEST()
209 graph.AddTensors(2); in TEST()
231 graph.AddTensors(3); in TEST()
258 graph.AddTensors(3); in TEST()
286 graph.AddTensors(3); in TEST()
316 graph.AddTensors(4); in TEST()
Dinterpreter_test.cc93 ASSERT_EQ(interpreter.AddTensors(1, &tensor_index), kTfLiteOk); in TEST()
111 ASSERT_EQ(interpreter.AddTensors(1, &tensor_index), kTfLiteOk); in TEST()
134 ASSERT_EQ(interpreter.AddTensors(2, &base_index), kTfLiteOk); in TEST()
137 ASSERT_EQ(interpreter.AddTensors(3, &base_index), kTfLiteOk); in TEST()
139 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
176 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
200 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
217 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
290 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
326 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.cc298 TfLiteStatus Interpreter::AddTensors(int tensors_to_add, in AddTensors() function in tflite::Interpreter
300 return primary_subgraph().AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc105 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAddSubgraph()
136 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildMulSubgraph()
167 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadSubgraph()
198 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildIfSubgraph()
235 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildLessEqualCondSubgraph()
272 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAccumulateLoopBodySubgraph()
319 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildPadLoopBodySubgraph()
366 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildWhileSubgraph()
400 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildAssignRandomValueToVariableSubgraph()
427 ASSERT_EQ(subgraph->AddTensors(kTensorCount, &first_new_tensor_index), in BuildCallOnceAndReadVariableSubgraph()
[all …]
Dconv.cc252 context->AddTensors(context, 1, &data->im2col_id); in AllocateTemporaryTensorsIfRequired()
259 context->AddTensors(context, 1, &data->hwcn_weights_id); in AllocateTemporaryTensorsIfRequired()
269 context, context->AddTensors(context, 1, &data->input_quantized_id)); in AllocateTemporaryTensorsIfRequired()
278 context, context->AddTensors(context, 1, &data->scaling_factors_id)); in AllocateTemporaryTensorsIfRequired()
286 context, context->AddTensors(context, 1, &data->accum_scratch_id)); in AllocateTemporaryTensorsIfRequired()
293 context, context->AddTensors(context, 1, &data->input_offset_id)); in AllocateTemporaryTensorsIfRequired()
300 context->AddTensors(context, 1, &data->row_sums_id)); in AllocateTemporaryTensorsIfRequired()
Dvariable_ops_test.cc54 ASSERT_EQ(interpreter_.AddTensors(4, &first_new_tensor_index), kTfLiteOk); in ConstructGraph()
Ddepthwise_conv.cc211 context, context->AddTensors(context, 1, &data->input_quantized_id)); in Prepare()
217 context, context->AddTensors(context, 1, &data->scaling_factors_id)); in Prepare()
223 context, context->AddTensors(context, 1, &data->input_offset_id)); in Prepare()
Dtranspose_conv.cc133 context->AddTensors(context, 1, &data->col2im_id); in AllocateTemporaryTensorsIfRequired()
144 context->AddTensors(context, 1, &data->transposed_weights_id); in AllocateTemporaryTensorsIfRequired()
155 context->AddTensors(context, 1, &data->scratch_tensor_id); in AllocateTemporaryTensorsIfRequired()
Ddetection_postprocess.cc124 context->AddTensors(context, 1, &op_data->decoded_boxes_index); in Init()
125 context->AddTensors(context, 1, &op_data->scores_index); in Init()
126 context->AddTensors(context, 1, &op_data->active_candidate_index); in Init()
Dnumeric_verify.cc102 context, context->AddTensors(context, 1, &op_data->cache_tensor_id)); in Prepare()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc61 interpreter.AddTensors(3); in TEST_P()
104 interpreter.AddTensors(3); in TEST_P()
138 interpreter.AddTensors(4); in TEST_P()
185 interpreter.AddTensors(5); in TEST_P()
240 interpreter.AddTensors(3); in TEST_P()
280 interpreter.AddTensors(3); in TEST_P()
323 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/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_ops_test.cc331 void AddTensors(bool table_two_initialization = false) { in AddTensors() function in tflite::__anon861f49270111::HashtableGraph
334 ASSERT_EQ(interpreter_->AddTensors(7, &first_new_tensor_index), in AddTensors()
343 ASSERT_EQ(interpreter_->AddTensors(14, &first_new_tensor_index), in AddTensors()
522 graph_->AddTensors(); in HashtableDefaultGraphTest()
583 graph.AddTensors(); in TEST()
596 graph.AddTensors(); in TEST()
612 graph.AddTensors(/*table_two_initialization=*/true); in TEST()
628 graph.AddTensors(); in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc194 EXPECT_EQ(interpreter_.AddTensors(5), kTfLiteOk); in InterpreterFp16()
467 EXPECT_EQ(interpreter_.AddTensors(4), kTfLiteOk); in InterpreterFp32()
582 EXPECT_EQ(interpreter_.AddTensors(8), kTfLiteOk); in Interpreter2Fp32()
762 EXPECT_EQ(interpreter_.AddTensors(8), kTfLiteOk); in InterpreterMultiNode()
1082 EXPECT_EQ(interpreter_.AddTensors(6), kTfLiteOk); in InterpreterQuantized()
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc31 interpreter_->AddTensors(5); in SetUp()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h95 TfLiteStatus AddTensors(int tensors_to_add,
486 static TfLiteStatus AddTensors(TfLiteContext* context, int tensors_to_add,
Dsubgraph.cc223 context_.AddTensors = AddTensors; in Subgraph()
1151 TfLiteStatus Subgraph::AddTensors(int tensors_to_add, in AddTensors() function in tflite::Subgraph
1165 TfLiteStatus Subgraph::AddTensors(TfLiteContext* context, int tensors_to_add, in AddTensors() function in tflite::Subgraph
1171 ->AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc98 subgraph->AddTensors(5); in SetUpSubgraph()
333 interpreter_->AddTensors(13); in SetUp()
Dutils_test.cc37 context.AddTensors = [](struct TfLiteContext*, int tensors_to_add, in TEST()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_allocator.cc171 TfLiteStatus AddTensors(const SubGraph* subgraph,
190 TfLiteStatus AllocationInfoBuilder::AddTensors(const SubGraph* subgraph, in AddTensors() function in tflite::__anon679442d00111::AllocationInfoBuilder
1062 builder.AddTensors(subgraph, offline_planner_offsets, eval_tensors)); in CommitStaticMemoryPlan()

12