| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv_test.cc | 456 m.SetNumThreads(num_thread); in SimpleDilatedTestPaddingSame()
514 m.SetNumThreads(num_thread); in BatchPaddingValidTest()
570 m.SetNumThreads(num_thread); in BatchPaddingSameTest()
954 m.SetNumThreads(4); in TEST_P()
1038 m.SetNumThreads(4); in TEST_P()
1125 m.SetNumThreads(4); in TEST_P()
1192 m.SetNumThreads(3); in TEST_P()
1265 m.SetNumThreads(2); in TEST_P()
1329 m.SetNumThreads(3); in TEST_P()
1401 m.SetNumThreads(4); in TEST_P()
[all …]
|
| D | eigen_support.cc | 97 SetNumThreads(num_threads); in LazyEigenThreadPoolHolder()
112 void SetNumThreads(int num_threads) { in SetNumThreads() function in tflite::eigen_support::__anon97ef62d90111::LazyEigenThreadPoolHolder
146 ptr->thread_pool_holder->SetNumThreads(context->recommended_num_threads); in Refresh()
|
| D | conv_test.cc | 565 m.SetNumThreads(i); in TEST_P()
644 m.SetNumThreads(i); in TEST_P()
941 m.SetNumThreads(1); in TEST_P()
944 m.SetNumThreads(2); in TEST_P()
947 m.SetNumThreads(3); in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | meta_support.h | 38 void SetNumThreads(int num_threads);
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_test.cc | 267 interpreter_->SetNumThreads(1); in TEST_F()
292 interpreter_->SetNumThreads(4); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.h | 89 TfLiteStatus SetNumThreads(int num_threads);
|
| D | model_test.cc | 312 ASSERT_EQ(builder.SetNumThreads(42), kTfLiteOk); in TEST()
317 ASSERT_EQ(builder.SetNumThreads(0), kTfLiteOk); in TEST()
322 ASSERT_EQ(builder.SetNumThreads(-1), kTfLiteOk); in TEST()
328 ASSERT_EQ(builder.SetNumThreads(-2), kTfLiteError); in TEST()
|
| D | interpreter_builder.cc | 671 TfLiteStatus InterpreterBuilder::SetNumThreads(int num_threads) { in SetNumThreads() function in tflite::InterpreterBuilder
684 TfLiteStatus status = SetNumThreads(num_threads); in operator ()()
744 (*interpreter)->SetNumThreads(num_threads_); in operator ()()
|
| D | interpreter_test.cc | 1128 ASSERT_EQ(interpreter_.SetNumThreads(4), kTfLiteOk); in TEST_F()
1132 ASSERT_EQ(interpreter_.SetNumThreads(4), kTfLiteOk); in TEST_F()
1133 ASSERT_EQ(interpreter_.SetNumThreads(5), kTfLiteOk); in TEST_F()
1139 ASSERT_EQ(interpreter_.SetNumThreads(-2), kTfLiteError); in TEST_F()
1142 ASSERT_EQ(interpreter_.SetNumThreads(-1), kTfLiteOk); in TEST_F()
1147 ASSERT_EQ(interpreter_.SetNumThreads(4), kTfLiteOk); in TEST_F()
1151 ASSERT_EQ(interpreter_.SetNumThreads(0), kTfLiteOk); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/ |
| D | interpreter_utils.cc | 48 interpreter->SetNumThreads(1); in InterpreterInvokeWithOpResolver()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.h | 102 PyObject* SetNumThreads(int num_threads);
|
| D | interpreter_wrapper_pybind11.cc | 218 return tensorflow::PyoOrThrow(self.SetNumThreads(num_threads)); in PYBIND11_MODULE()
|
| /external/XNNPACK/bench/ |
| D | convert.cc | 489 interpreter->SetNumThreads(1); in tflite_convert_f16_f32()
588 interpreter->SetNumThreads(1); in tflite_convert_f32_qs8()
689 interpreter->SetNumThreads(1); in tflite_convert_f32_qu8()
792 interpreter->SetNumThreads(1); in tflite_convert_qs8_f32()
895 interpreter->SetNumThreads(1); in tflite_convert_qu8_f32()
|
| D | sigmoid.cc | 296 interpreter->SetNumThreads(1); in tflite_sigmoid_f32()
402 interpreter->SetNumThreads(1); in tflite_sigmoid_qs8()
508 interpreter->SetNumThreads(1); in tflite_sigmoid_qu8()
|
| D | elu.cc | 225 interpreter->SetNumThreads(1); in tflite_elu_f32()
331 interpreter->SetNumThreads(1); in tflite_elu_qs8()
|
| D | abs.cc | 154 interpreter->SetNumThreads(1); in tflite_abs_f32()
|
| D | square.cc | 154 interpreter->SetNumThreads(1); in tflite_square_f32()
|
| D | square-root.cc | 154 interpreter->SetNumThreads(1); in tflite_square_root_f32()
|
| D | bankers-rounding.cc | 154 interpreter->SetNumThreads(1); in tflite_bankers_rounding_f32()
|
| D | floor.cc | 154 interpreter->SetNumThreads(1); in tflite_floor_f32()
|
| D | negate.cc | 154 interpreter->SetNumThreads(1); in tflite_negate_f32()
|
| D | leaky-relu.cc | 159 interpreter->SetNumThreads(1); in tflite_leaky_relu_f32()
|
| D | ceiling.cc | 154 interpreter->SetNumThreads(1); in tflite_ceiling_f32()
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | tflite_inference_stage.cc | 118 interpreter_->SetNumThreads(params.num_threads()); in Init()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api.cc | 285 interpreter->SetNumThreads(optional_options->num_threads); in InterpreterCreateWithOpResolver()
|