Searched refs:AddTensors (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | graph_info_test.cc | 57 void AddTensors(int count) { tensors_.resize(count + tensors_.size()); } in AddTensors() function in tflite::__anonc155f4c20111::SimpleTestGraph
114 graph.AddTensors(2); in TEST()
134 graph.AddTensors(1); in TEST()
154 graph.AddTensors(2); in TEST()
176 graph.AddTensors(3); in TEST()
204 graph.AddTensors(3); in TEST()
234 graph.AddTensors(4); in TEST()
|
| D | interpreter_test.cc | 49 ASSERT_EQ(interpreter.AddTensors(2, &base_index), kTfLiteOk); in TEST()
52 ASSERT_EQ(interpreter.AddTensors(3, &base_index), kTfLiteOk); in TEST()
54 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
91 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
115 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
150 ASSERT_EQ(interpreter.AddTensors(2), kTfLiteOk); in TEST()
189 ASSERT_EQ(interpreter.AddTensors(4), kTfLiteOk); in TEST()
206 ASSERT_EQ(interpreter.AddTensors(10), kTfLiteOk); in TEST()
256 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
273 ASSERT_EQ(interpreter.AddTensors(1), kTfLiteOk); in TEST()
[all …]
|
| D | string_util_test.cc | 26 interpreter.AddTensors(3); in TEST()
78 interpreter.AddTensors(1); in TEST()
101 interpreter.AddTensors(1); in TEST()
|
| D | interpreter.cc | 76 context_.AddTensors = AddTensors; in Interpreter()
448 TfLiteStatus Interpreter::AddTensors(int tensors_to_add, in AddTensors() function in tflite::Interpreter
461 TfLiteStatus Interpreter::AddTensors(TfLiteContext* context, int tensors_to_add, in AddTensors() function in tflite::Interpreter
467 ->AddTensors(tensors_to_add, first_new_tensor_index); in AddTensors()
|
| D | interpreter.h | 129 TfLiteStatus AddTensors(int tensors_to_add,
336 static TfLiteStatus AddTensors(TfLiteContext* context, int tensors_to_add,
|
| D | context.h | 277 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, member
|
| D | model.cc | 778 if ((**interpreter).AddTensors(tensors->Length()) != kTfLiteOk) { in operator ()()
|
| /external/tensorflow/tensorflow/contrib/lite/examples/label_image/ |
| D | bitmap_helpers_impl.h | 46 interpreter->AddTensors(2, &base_index); in resize()
48 interpreter->AddTensors(1, &base_index); in resize()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | conv.cc | 83 context->AddTensors(context, 1, &data->im2col_id); in Init()
84 context->AddTensors(context, 1, &data->hwcn_weights_id); in Init()
|
| D | mean.cc | 51 context->AddTensors(context, 2, scratch_tensor_index); in Init()
|
| D | svdf.cc | 44 context->AddTensors(context, 1, scratch_tensor_index); in Init()
|