/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
D | egl_environment.cc | 92 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()
|
D | compiler.cc | 97 : 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
|
D | egl_environment.h | 47 const GpuInfo& gpu_info() const { return gpu_info_; } in gpu_info() 59 GpuInfo gpu_info_; variable
|
D | api2.cc | 502 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
|
D | api.cc | 197 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
|
D | runtime.h | 85 const GpuInfo gpu_info_; variable
|
D | runtime.cc | 244 gpu_info_(gpu_info), in Runtime()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/java/src/main/native/ |
D | gpu_delegate_jni.cc | 73 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/ |
D | calculator.cc | 62 : gpu_info_{gpu_info} {} in WorkgroupsCalculator() 69 MaybeShrinkWorkgroup(gpu_info_, &workgroup_size); in Calculate()
|
D | calculator.h | 51 GpuInfo gpu_info_;
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
D | converter.cc | 45 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
|