| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 47 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in InspectNumaNodes()
71 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in DeviceToBus()
83 return GetExecutor(config).ValueOrDie(); in FirstExecutorForBus()
117 return GetExecutor(config); in ExecutorForDevice()
126 return GetExecutor(config); in ExecutorForDeviceWithPluginConfig()
129 port::StatusOr<StreamExecutor*> ROCmPlatform::GetExecutor( in GetExecutor() function in stream_executor::gpu::ROCmPlatform
|
| D | rocm_platform.h | 75 port::StatusOr<StreamExecutor*> GetExecutor(
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 325 StreamExecutor* GetExecutor(int ordinal) { in GetExecutor() function in stream_executor::__anon5e0d9c3a0111::StreamExecutorTest
358 StreamExecutor* executor = GetExecutor(0); in TEST_F()
377 StreamExecutor* executor = GetExecutor(0); in TEST_F()
399 StreamExecutor* executor = GetExecutor(0); in TEST_F()
417 StreamExecutor* executor = GetExecutor(0); in TEST_F()
433 StreamExecutor* executor = GetExecutor(0); in TEST_F()
457 StreamExecutor* executor = GetExecutor(0); in TEST_F()
477 StreamExecutor* executor = GetExecutor(0); in TEST_F()
498 StreamExecutor* executor = GetExecutor(0); in TEST_F()
525 StreamExecutor* executor = GetExecutor(0); in TEST_F()
[all …]
|
| D | stream_executor_internal.h | 68 port::StatusOr<StreamExecutor*> GetExecutor(
|
| D | stream_executor.cc | 709 return GetExecutor(config); in ExecutorForDevice()
716 return GetExecutor(config); in ExecutorForDeviceWithPluginConfig()
718 port::StatusOr<StreamExecutor*> CPlatform::GetExecutor( in GetExecutor() function in stream_executor::CPlatform
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 85 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in InspectNumaNodes()
109 StreamExecutor* exec = GetExecutor(config).ValueOrDie(); in DeviceToBus()
121 return GetExecutor(config).ValueOrDie(); in FirstExecutorForBus()
154 return GetExecutor(config); in ExecutorForDevice()
163 return GetExecutor(config); in ExecutorForDeviceWithPluginConfig()
166 port::StatusOr<StreamExecutor*> CudaPlatform::GetExecutor( in GetExecutor() function in stream_executor::gpu::CudaPlatform
|
| D | cuda_platform.h | 75 port::StatusOr<StreamExecutor*> GetExecutor(
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 54 return GetExecutor(config); in ExecutorForDevice()
63 return GetExecutor(config); in ExecutorForDeviceWithPluginConfig()
66 port::StatusOr<StreamExecutor*> HostPlatform::GetExecutor( in GetExecutor() function in stream_executor::host::HostPlatform
|
| D | host_platform.h | 62 port::StatusOr<StreamExecutor*> GetExecutor(
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 56 return GetExecutor(config); in ExecutorForDevice()
66 return GetExecutor(config); in ExecutorForDeviceWithPluginConfig()
69 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::GetExecutor( in GetExecutor() function in stream_executor::interpreter::XlaInterpreterPlatform
|
| D | platform.h | 51 port::StatusOr<StreamExecutor*> GetExecutor(
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | pool_allocator_test.cc | 34 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()
54 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()
89 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()
150 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()
256 platform->GetExecutor(se::StreamExecutorConfig(/*ordinal=*/0)) in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_platform.h | 89 return GetExecutor(config); in ExecutorForDevice()
99 return GetExecutor(config); in ExecutorForDeviceWithPluginConfig()
102 StatusOr<::stream_executor::StreamExecutor*> GetExecutor(
|
| D | tpu_platform.cc | 81 StatusOr<::stream_executor::StreamExecutor*> TpuPlatform::GetExecutor( in GetExecutor() function in tensorflow::tpu::TpuPlatform
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | cpu_device.cc | 52 platform->GetExecutor(config)); in GetCpuClient()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.h | 164 virtual port::StatusOr<StreamExecutor*> GetExecutor(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | platform_util.cc | 203 auto executor_status = platform->GetExecutor(config); in GetStreamExecutors()
|
| /external/tensorflow/tensorflow/core/platform/s3/ |
| D | s3_file_system.h | 123 std::shared_ptr<Aws::Utils::Threading::PooledThreadExecutor> GetExecutor();
|
| D | s3_file_system.cc | 548 this->GetExecutor().get()); in GetTransferManager()
561 S3FileSystem::GetExecutor() { in GetExecutor() function in tensorflow::S3FileSystem
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 324 se::port::StatusOr<se::StreamExecutor*> GetExecutor( in TEST() function in tensorflow::__anonde1592d50111::TEST::FakePlatform
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver_test.cc | 91 platform->GetExecutor(config)); in GetCpuClientWithNonLocalDevice()
|
| /external/tensorflow/tensorflow/c/experimental/filesystem/plugins/s3/ |
| D | s3_filesystem.cc | 221 static void GetExecutor(tf_s3_filesystem::S3File* s3_file) { in GetExecutor() function
236 GetExecutor(s3_file); in GetTransferManager()
|