Searched refs:gpu_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 150 gpu_.reset(reinterpret_cast<BaseGPUDevice*>(device_.release())); in GPUDeviceTestHelper()
156 BaseGPUDevice* gpu() { return gpu_.get(); } in gpu()
159 se::Stream* compute_stream() { return gpu_->stream_->compute; } in compute_stream()
160 se::Stream* h2d_stream() { return gpu_->stream_->host_to_device; } in h2d_stream()
161 se::Stream* d2h_stream() { return gpu_->stream_->device_to_host; } in d2h_stream()
162 se::Stream* d2d_stream() { return gpu_->stream_->device_to_device[0]; } in d2d_stream()
163 EventMgr* event_mgr() { return gpu_->em_; } in event_mgr()
164 int pending_cap() { return gpu_->pending_cap_; } in pending_cap()
168 std::unique_ptr<BaseGPUDevice> gpu_; member in tensorflow::GPUDeviceTestHelper
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | BUILD | 696 name = "gpu_" + op + "_lib",
775 name = "gpu_" + op + "_lib",
814 name = "gpu_" + op + "_lib",
830 name = "gpu_" + op + "_lib",
854 name = "gpu_" + op + "_lib",
879 name = "gpu_" + op + "_lib",
901 name = "gpu_" + op + "_lib",
937 name = "gpu_" + op + "_lib",
965 name = "gpu_" + op + "_lib",
986 name = "gpu_" + op + "_lib",
|