| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.cc | 113 StreamExecutor *from = ExecutorForDevice(i).ValueOrDie(); in GetPeerAccessMap()
114 StreamExecutor *to = ExecutorForDevice(j).ValueOrDie(); in GetPeerAccessMap()
127 StreamExecutor *from = ExecutorForDevice(devices.first).ValueOrDie(); in EnablePeerAccess()
128 StreamExecutor *to = ExecutorForDevice(devices.second).ValueOrDie(); in EnablePeerAccess()
|
| D | platform.h | 154 virtual port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) = 0;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 55 StreamExecutor* stream_exec = platform->ExecutorForDevice(0).ValueOrDie(); in TEST()
130 StreamExecutor* stream_exec = platform->ExecutorForDevice(0).ValueOrDie(); in TEST()
|
| D | memcpy_test.cc | 29 StreamExecutor* executor = platform->ExecutorForDevice(0).ValueOrDie(); in TEST()
|
| D | cuda_platform.h | 70 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
|
| D | cuda_platform.cc | 149 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::gpu::CudaPlatform
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_stream_test.cc | 29 se::StreamExecutor* executor = platform->ExecutorForDevice(0).ValueOrDie(); in TEST()
|
| D | host_platform.h | 57 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
|
| D | host_platform.cc | 49 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::host::HostPlatform
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_id_utils.h | 36 return device_manager->ExecutorForDevice(platform_device_id.value()); in ExecutorForPlatformDeviceId()
|
| D | device_event_mgr_test.cc | 112 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()
121 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()
443 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in BM_no_ops()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.h | 46 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
|
| D | platform.cc | 50 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::ExecutorForDevice( in ExecutorForDevice() function in stream_executor::interpreter::XlaInterpreterPlatform
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.h | 70 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
|
| D | rocm_platform.cc | 112 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::gpu::ROCmPlatform
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 92 se::StreamExecutor* se = gpu_platform->ExecutorForDevice(gpu_id).ValueOrDie(); in AvailableGPUMemory()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | transfer_ops.cc | 84 tpu_platform->ExecutorForDevice(real_device_ordinal)); in RunTransferWithOrdinal()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_platform.h | 85 StatusOr<::stream_executor::StreamExecutor*> ExecutorForDevice( in ExecutorForDevice() function
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_internal.h | 65 port::StatusOr<StreamExecutor*> ExecutorForDevice(int ordinal) override;
|
| D | stream_executor_test.cc | 228 platform->ExecutorForDevice(0); in TEST()
332 cplatform_->ExecutorForDevice(ordinal); in GetExecutor()
|
| D | stream_executor.cc | 706 port::StatusOr<StreamExecutor*> CPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::CPlatform
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator_test.cc | 36 stream_exec_(platform_->ExecutorForDevice(0).ValueOrDie()) {} in BufferComparatorTest()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 660 local_client_->platform()->ExecutorForDevice(d).ValueOrDie(); in XLA_TEST_F()
683 se::Stream wrong_stream(wrong_platform->ExecutorForDevice(0).ValueOrDie()); in XLA_TEST_F()
722 ->ExecutorForDevice(local_client_->default_device_ordinal()) in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 314 se::port::StatusOr<se::StreamExecutor*> ExecutorForDevice( in TEST() function in tensorflow::__anonde1592d50111::TEST::FakePlatform
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 308 {client->platform()->ExecutorForDevice(0).ValueOrDie()}); in ReplayComputation()
|