Searched refs:executor_ (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinfeed_manager.h50 : executor_(executor), length_(length) {
51 device_memory_ = executor_->AllocateArray<uint8>(length);
55 ~InfeedBuffer() { executor_->Deallocate(&device_memory_); }
68 perftools::gputools::StreamExecutor* executor_; // Not owned. member in class:xla::gpu::InfeedBuffer
/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.h96 gpu::StreamExecutor* executor() const { return executor_; }
102 gpu::StreamExecutor* executor_; // not owned member in class:tensorflow::BaseGPUDevice
H A Dgpu_device.cc278 executor_ = executor_status.ValueOrDie();
279 em_.reset(new EventMgr(executor_, options.config.gpu_options()));
288 StreamGroupFactory::Global().GetOrCreate(tf_gpu_id_, i, executor_));
303 bool ok = executor_->SynchronousMemZero(

Completed in 187 milliseconds