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.cc55 : command_queue_(command_queue) {} in OpenGlConverterImpl()
73 if (command_queue_) { in Dispatch()
74 return command_queue_->Dispatch(program_, num_workgroups); in Dispatch()
81 CommandQueue* command_queue_; member in tflite::gpu::gl::__anon1e444b000111::OpenGlConverterImpl
350 : command_queue_(command_queue) {} in TensorConverterBuilderImpl()
378 impl = std::make_unique<FromTensorConverter>(command_queue_); in MakeConverter()
380 impl = std::make_unique<ToTensorConverter>(command_queue_); in MakeConverter()
390 CommandQueue* command_queue_; member in tflite::gpu::gl::__anon1e444b000111::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::__anon754217700111::Delegate
Dmetal_delegate.mm109 command_queue_ = command_queue;
110 device_ = [command_queue_ device];
128 id<MTLCommandBuffer> command_buffer = [command_queue_ commandBuffer];
154 id<MTLCommandQueue> command_queue_;
183 command_queue_ = [metal_device_ newCommandQueue];
185 gpu_alarm_clock_ = std::unique_ptr<GpuAlarmClock>(new GpuAlarmClock(command_queue_));
479 command_buffer = [command_queue_ commandBuffer];
512 inference_context_.EncodeWithCommandQueue(command_queue_, flush_period);
513 command_buffer = [command_queue_ commandBuffer];
548 id<MTLCommandBuffer> signal_cb = [command_queue_ commandBuffer];
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Druntime.h63 CommandQueue* command_queue() { return command_queue_; } in command_queue()
88 CommandQueue* command_queue_; variable
Druntime.cc249 command_queue_(command_queue) { in Runtime()
653 RETURN_IF_ERROR(command_queue_->Dispatch(descriptor.program, in Execute()