Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dconverter.cc50 : command_queue_(command_queue) {} in OpenGlConverterImpl()
68 if (command_queue_) { in Dispatch()
69 return command_queue_->Dispatch(program_, num_workgroups); in Dispatch()
76 CommandQueue* command_queue_; member in tflite::gpu::gl::__anon70b837870111::OpenGlConverterImpl
345 : command_queue_(command_queue) {} in TensorConverterBuilderImpl()
373 impl = absl::make_unique<FromTensorConverter>(command_queue_); in MakeConverter()
375 impl = absl::make_unique<ToTensorConverter>(command_queue_); in MakeConverter()
385 CommandQueue* command_queue_; member in tflite::gpu::gl::__anon70b837870111::TensorConverterBuilderImpl
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc263 command_queue_ = NewCommandQueue(gpu_info); in Prepare()
273 command_queue_.get(), in Prepare()
295 ref.shape, *external_object, command_queue_.get(), in Invoke()
320 command_queue_.get(), external_object)); in Invoke()
327 RETURN_IF_ERROR(command_queue_->WaitForCompletion()); in Invoke()
368 std::unique_ptr<CommandQueue> command_queue_; member in tflite::gpu::gl::__anon3ca958170111::Delegate
Dmetal_delegate.mm108 command_queue_ = command_queue;
109 device_ = [command_queue_ device];
127 id<MTLCommandBuffer> command_buffer = [command_queue_ commandBuffer];
153 id<MTLCommandQueue> command_queue_;
182 command_queue_ = [metal_device_ newCommandQueue];
184 gpu_alarm_clock_ = std::unique_ptr<GpuAlarmClock>(new GpuAlarmClock(command_queue_));
454 command_buffer = [command_queue_ commandBuffer];
491 inference_context_.EncodeWithCommandQueue(command_queue_, flush_period);
492 command_buffer = [command_queue_ commandBuffer];
526 id<MTLCommandBuffer> signal_cb = [command_queue_ commandBuffer];
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Druntime.h62 CommandQueue* command_queue() { return command_queue_; } in command_queue()
87 CommandQueue* command_queue_; variable
Druntime.cc246 command_queue_(command_queue) { in Runtime()
650 RETURN_IF_ERROR(command_queue_->Dispatch(descriptor.program, in Execute()