Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
Dinternal_api_samples.cc156 InferenceEnvironmentOptions env_options; in RunModelSampleWithInternalAPI() local
157 env_options.device = env.device().id(); in RunModelSampleWithInternalAPI()
158 env_options.context = env.context().context(); in RunModelSampleWithInternalAPI()
159 env_options.command_queue = env.queue()->queue(); in RunModelSampleWithInternalAPI()
160 RETURN_IF_ERROR(NewInferenceEnvironment(env_options, &inf_env, nullptr)); in RunModelSampleWithInternalAPI()
280 InferenceEnvironmentOptions env_options; in RunModelSampleWithInternalAPISerializedKernels() local
281 env_options.device = env.device().id(); in RunModelSampleWithInternalAPISerializedKernels()
282 env_options.context = env.context().context(); in RunModelSampleWithInternalAPISerializedKernels()
283 env_options.command_queue = env.queue()->queue(); in RunModelSampleWithInternalAPISerializedKernels()
284 env_options.serialized_binary_cache = in RunModelSampleWithInternalAPISerializedKernels()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dgpu_api_delegate.cc104 InferenceEnvironmentOptions env_options; in Prepare() local
105 env_options.egl_context = options_.egl_context; in Prepare()
106 env_options.egl_display = options_.egl_display; in Prepare()
107 env_options.serialized_binary_cache = { in Prepare()
112 NewInferenceEnvironment(env_options, &environment_, &properties); in Prepare()
Dapi.cc653 const InferenceEnvironmentOptions& env_options, in Initialize() argument
669 if (env_options.IsGlAware() && in Initialize()
672 env_options.egl_display, environment_); in Initialize()
686 absl::Status Initialize(const InferenceEnvironmentOptions& env_options, in Initialize() argument
695 if (env_options.IsGlAware() && in Initialize()
698 env_options.egl_display, environment_); in Initialize()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Ddelegate.cc290 cl::InferenceEnvironmentOptions env_options; in InitializeOpenClApi() local
292 RETURN_IF_ERROR(cl::NewInferenceEnvironment(env_options, &cl_environment_, in InitializeOpenClApi()
322 gl::InferenceEnvironmentOptions env_options; in InitializeOpenGlApi() local
325 NewInferenceEnvironment(env_options, &gl_environment_, &properties)); in InitializeOpenGlApi()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc366 explicit TensorTieFactory(const InferenceEnvironmentOptions& env_options) in TensorTieFactory() argument
367 : converter_builder_(NewConverterBuilder(env_options.queue)) {} in TensorTieFactory()
496 InferenceBuilderImpl(const InferenceEnvironmentOptions& env_options, in InferenceBuilderImpl() argument
499 : env_options_(env_options), in InferenceBuilderImpl()