Home
last modified time | relevance | path

Searched refs:gpu_settings (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
Dgpu_plugin.cc37 const auto* gpu_settings = tflite_settings.gpu_settings(); in GpuPlugin() local
38 if (gpu_settings) { in GpuPlugin()
40 gpu_settings->is_precision_loss_allowed() in GpuPlugin()
43 if (gpu_settings->enable_quantized_inference()) { in GpuPlugin()
47 if (gpu_settings->force_backend() == GPUBackend_OPENCL) { in GpuPlugin()
49 } else if (gpu_settings->force_backend() == GPUBackend_OPENGL) { in GpuPlugin()
Dconfiguration_generated.h1425 std::unique_ptr<tflite::GPUSettingsT> gpu_settings; member
1459 const tflite::GPUSettings *gpu_settings() const { in gpu_settings() function
1489 verifier.VerifyTable(gpu_settings()) && in Verify()
1519 void add_gpu_settings(flatbuffers::Offset<tflite::GPUSettings> gpu_settings) { in add_gpu_settings()
1520 fbb_.AddOffset(TFLiteSettings::VT_GPU_SETTINGS, gpu_settings); in add_gpu_settings()
1559 flatbuffers::Offset<tflite::GPUSettings> gpu_settings = 0,
1575 builder_.add_gpu_settings(gpu_settings);
2475 …{ auto _e = gpu_settings(); if (_e) _o->gpu_settings = std::unique_ptr<tflite::GPUSettingsT>(_e->U… in UnPackTo()
2495 …auto _gpu_settings = _o->gpu_settings ? CreateGPUSettings(_fbb, _o->gpu_settings.get(), _rehasher)… in CreateTFLiteSettings()
Dproto_to_flatbuffer.cc270 ConvertGPUSettings(settings.gpu_settings(), builder), in ConvertTfliteSettings()
Dconfiguration.proto306 optional GPUSettings gpu_settings = 3; field
/external/tensorflow/tensorflow/lite/delegates/
Dtelemetry_test.cc71 EXPECT_EQ(settings->gpu_settings()->is_precision_loss_allowed(), in BeginEvent()
118 flatbuffers::Offset<tflite::GPUSettings> gpu_settings = in TEST() local
126 /*gpu_settings=*/gpu_settings)); in TEST()