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::__anonfbcd0ba20111::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::__anonfbcd0ba20111::TensorConverterBuilderImpl
/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()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dgl_delegate.cc264 command_queue_ = NewCommandQueue(gpu_info); in Prepare()
274 command_queue_.get(), in Prepare()
296 ref.shape, *external_object, command_queue_.get(), in Invoke()
321 command_queue_.get(), external_object)); in Invoke()
328 RETURN_IF_ERROR(command_queue_->WaitForCompletion()); in Invoke()
369 std::unique_ptr<CommandQueue> command_queue_; member in tflite::gpu::gl::__anon640f09d20111::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_));
456 command_buffer = [command_queue_ commandBuffer];
493 inference_context_.EncodeWithCommandQueue(command_queue_, flush_period);
494 command_buffer = [command_queue_ commandBuffer];
528 id<MTLCommandBuffer> signal_cb = [command_queue_ commandBuffer];
[all …]