Home
last modified time | relevance | path

Searched refs:gpu_info_ (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Degl_environment.cc92 if (gpu_info_.vendor == GpuVendor::kUnknown) { in Init()
93 RETURN_IF_ERROR(RequestGpuInfo(&gpu_info_)); in Init()
112 RETURN_IF_ERROR(RequestGpuInfo(&gpu_info_)); in InitSurfacelessContext()
113 if (gpu_info_.IsPowerVR()) { in InitSurfacelessContext()
Dcompiler.cc97 : node_shader_(*node_shader), gpu_info_(*gpu_info), options_(options) { in CompilerImpl()
125 NodeShader::GenerationContext ctx = {&gpu_info_, options_, in Compile()
206 if (num_textures < gpu_info_.GetMaxImageArguments() && in Compile()
207 !ExceedsMaxSize(*object, gpu_info_) && in Compile()
235 ShaderCodegen codegen(options_, gpu_info_); in Compile()
292 const GpuInfo& gpu_info_; member in tflite::gpu::gl::__anon275ffe220111::CompilerImpl
Degl_environment.h47 const GpuInfo& gpu_info() const { return gpu_info_; } in gpu_info()
59 GpuInfo gpu_info_; variable
Dapi2.cc502 gpu_info_(gpu_info), in InferenceBuilderImpl()
570 auto compiler = NewCompiler(kernels.get(), gpu_info_, compiler_options); in Build()
571 auto workgroup_calculator = NewDefaultWorkgroupsCalculator(*gpu_info_); in Build()
577 absl::make_unique<Runtime>(runtime_options, *gpu_info_, in Build()
653 const GpuInfo* gpu_info_; member in tflite::gpu::gl::__anon517d4d930111::InferenceBuilderImpl
667 RETURN_IF_ERROR(RequestGpuInfo(&gpu_info_)); in Init()
668 properties_.is_opengl_available = gpu_info_.IsApiOpenGl31OrAbove(); in Init()
674 queue_ = NewCommandQueue(gpu_info_); in Init()
693 env_options_, resolved_options, std::move(model), &gpu_info_); in NewInferenceBuilder()
707 GpuInfo gpu_info_; member in tflite::gpu::gl::__anon517d4d930111::InferenceEnvironmentImpl
Dapi.cc197 explicit CompiledModelImpl(const GpuInfo& gpu_info) : gpu_info_(gpu_info) {} in CompiledModelImpl()
262 auto runtime = absl::make_unique<Runtime>(options, gpu_info_, command_queue, in NewRun()
362 const GpuInfo gpu_info_; member in tflite::gpu::gl::__anone3705d010111::CompiledModelImpl
Druntime.h85 const GpuInfo gpu_info_; variable
Druntime.cc244 gpu_info_(gpu_info), in Runtime()
/external/tensorflow/tensorflow/lite/delegates/gpu/java/src/main/native/
Dgpu_delegate_jni.cc73 status = tflite::gpu::gl::RequestGpuInfo(&gpu_info_); in ReadInfo()
80 return compatibility_list_->Includes(android_info_, gpu_info_); in IsDelegateSupportedOnThisDevice()
85 tflite::gpu::GpuInfo gpu_info_; member in __anonbff7837d0111::CompatibilityListHelper
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/workgroups/
Dcalculator.cc62 : gpu_info_{gpu_info} {} in WorkgroupsCalculator()
69 MaybeShrinkWorkgroup(gpu_info_, &workgroup_size); in Calculate()
Dcalculator.h51 GpuInfo gpu_info_;
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dconverter.cc45 void SetGpuInfo(const GpuInfo& info) { gpu_info_ = info; } in SetGpuInfo()
57 GetPossibleWorkGroupsConv(TuningType::kFast, gpu_info_, kernel_.info_, grid, in DispatchKernel()
68 GpuInfo gpu_info_; member in tflite::gpu::cl::__anondaf6909e0111::OpenClConverterImpl