Searched defs:stream_exec (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 102 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
119 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
152 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
169 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
188 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
210 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
232 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
|
| D | gpu_debug_allocator_test.cc | 40 auto stream_exec = GpuIdUtil::ExecutorForCudaGpuId(cuda_gpu_id).ValueOrDie(); in TEST() local
127 auto stream_exec = GpuIdUtil::ExecutorForCudaGpuId(cuda_gpu_id).ValueOrDie(); in TEST() local
170 auto stream_exec = GpuIdUtil::ExecutorForCudaGpuId(cuda_gpu_id).ValueOrDie(); in TEST() local
|
| D | gpu_bfc_allocator.h | 55 explicit GPUMemAllocator(perftools::gputools::StreamExecutor* stream_exec) in GPUMemAllocator()
|
| D | pool_allocator.h | 184 explicit CUDAHostAllocator(perftools::gputools::StreamExecutor* stream_exec) in CUDAHostAllocator()
|
| D | gpu_device.cc | 1315 auto stream_exec = executor.ValueOrDie(); in GetValidDeviceIds() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | device_util.h | 32 string DeviceIdentifier(perftools::gputools::StreamExecutor* stream_exec) { in DeviceIdentifier()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_convolution_algorithm_picker.h | 37 perftools::gputools::StreamExecutor* stream_exec, in CudnnConvolutionAlgorithmPicker()
|
| D | gpu_compiler.cc | 132 se::StreamExecutor* stream_exec, in OptimizeHloModule()
470 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunHloPasses()
481 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_activation.cc | 35 StreamExecutor *stream_exec) in ScopedActivateExecutorContext()
|
| D | cuda_gpu_executor.cc | 116 CUDAExecutor *ExtractCudaExecutor(StreamExecutor *stream_exec) { in ExtractCudaExecutor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | event.cc | 25 Event::Event(StreamExecutor* stream_exec) in Event()
|
| D | stream_executor_pimpl.cc | 98 ScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in ScopedTracer()
139 MakeScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in MakeScopedTracer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 65 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 454 perftools::gputools::StreamExecutor* stream_exec, in RunBackend()
|