| /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::__anonf8036f140111::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/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 | 107 PyObject* SetNumThreads(int num_threads);
|
| D | interpreter_wrapper_pybind11.cc | 198 return tensorflow::PyoOrThrow(self.SetNumThreads(num_threads)); in PYBIND11_MODULE()
|
| D | interpreter_wrapper.cc | 752 PyObject* InterpreterWrapper::SetNumThreads(int num_threads) { in SetNumThreads() function in tflite::interpreter_wrapper::InterpreterWrapper
754 interpreter_->SetNumThreads(num_threads); in SetNumThreads()
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | tflite_inference_stage.cc | 118 interpreter_->SetNumThreads(params.num_threads()); in Init()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 1129 ASSERT_EQ(interpreter_.SetNumThreads(4), kTfLiteOk); in TEST_F()
1133 ASSERT_EQ(interpreter_.SetNumThreads(4), kTfLiteOk); in TEST_F()
1134 ASSERT_EQ(interpreter_.SetNumThreads(5), kTfLiteOk); in TEST_F()
1140 ASSERT_EQ(interpreter_.SetNumThreads(-2), kTfLiteError); in TEST_F()
1143 ASSERT_EQ(interpreter_.SetNumThreads(-1), kTfLiteOk); in TEST_F()
1148 ASSERT_EQ(interpreter_.SetNumThreads(4), kTfLiteOk); in TEST_F()
|
| D | interpreter.cc | 349 TfLiteStatus Interpreter::SetNumThreads(int num_threads) { in SetNumThreads() function in tflite::Interpreter
|
| D | interpreter.h | 447 TfLiteStatus SetNumThreads(int num_threads);
|
| /external/XNNPACK/bench/ |
| D | bankers-rounding.cc | 171 interpreter->SetNumThreads(1); in tflite_bankers_rounding_f32()
|
| D | elu.cc | 171 interpreter->SetNumThreads(1); in tflite_elu_f32()
|
| D | ceiling.cc | 171 interpreter->SetNumThreads(1); in tflite_ceiling_f32()
|
| D | square-root.cc | 171 interpreter->SetNumThreads(1); in tflite_square_root_f32()
|
| D | floor.cc | 171 interpreter->SetNumThreads(1); in tflite_floor_f32()
|
| D | prelu.cc | 194 interpreter->SetNumThreads(1); in tflite_prelu_f32()
|
| D | softmax.cc | 245 interpreter->SetNumThreads(1); in tflite_softmax_f32()
|
| D | sigmoid.cc | 246 interpreter->SetNumThreads(1); in tflite_sigmoid_f32()
|
| D | hardswish.cc | 173 interpreter->SetNumThreads(1); in tflite_hardswish_f32()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api.cc | 285 interpreter->SetNumThreads(optional_options->num_threads); in InterpreterCreateWithOpResolver()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | interpreter.py | 310 self._interpreter.SetNumThreads(num_threads)
|
| /external/tensorflow/tensorflow/lite/examples/ios/simple/ |
| D | RunModelViewController.mm | 120 interpreter->SetNumThreads(num_threads);
|