| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 100 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
117 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
150 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
167 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
186 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
208 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
230 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
249 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
|
| D | gpu_debug_allocator_test.cc | 43 auto stream_exec = in TEST() local
140 auto stream_exec = in TEST() local
187 auto stream_exec = in TEST() local
|
| D | gpu_host_allocator.h | 29 explicit GpuHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in GpuHostAllocator()
|
| D | gpu_bfc_allocator.h | 40 explicit GPUMemAllocator(se::StreamExecutor* stream_exec, in GPUMemAllocator()
|
| D | gpu_device.cc | 1534 auto stream_exec = executor.ValueOrDie(); in GetValidDeviceIds() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 113 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
139 std::vector<std::vector<se::StreamExecutor*>> stream_exec, in RunBackendOnModuleGroup()
147 std::vector<std::vector<se::StreamExecutor*>> stream_exec, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | device_util.h | 32 string DeviceIdentifier(se::StreamExecutor* stream_exec) { in DeviceIdentifier()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | event.cc | 24 Event::Event(StreamExecutor* stream_exec) in Event()
|
| D | stream_executor_pimpl.cc | 102 ScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in ScopedTracer()
143 MakeScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in MakeScopedTracer()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_activation.cc | 34 StreamExecutor* stream_exec) in ScopedActivateExecutorContext()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_algorithm_picker.h | 40 CudnnConvAlgorithmPicker(se::StreamExecutor* stream_exec, in CudnnConvAlgorithmPicker()
|
| D | buffer_comparator.cc | 84 auto stream_exec = stream->parent(); in Create() local
135 auto stream_exec = stream_->parent(); in CompareEqualImpl() local
|
| D | cudnn_conv_algorithm_picker.cc | 45 se::StreamExecutor* stream_exec) { in GetAlgorithms()
83 tensorflow::mutex_lock LockGpu(const se::StreamExecutor* stream_exec) { in LockGpu()
|
| D | nvptx_compiler.cc | 179 Status OptimizeHloModule(HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
619 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunHloPasses()
634 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
|
| D | cusolver_rewriter.cc | 202 CusolverRewriter::CusolverRewriter(se::StreamExecutor* stream_exec, in CusolverRewriter()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 60 se::StreamExecutor* stream_exec, in LogConvAutotuneResults()
83 se::StreamExecutor* stream_exec, in LogFusedConvAutotuneResults()
|
| D | conv_ops_gpu.h | 144 se::StreamExecutor* stream_exec) const { in ShouldIncludeWinogradNonfusedAlgo()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | llvm_compiler.cc | 35 std::vector<std::vector<se::StreamExecutor*>> stream_exec, in RunBackendOnModuleGroup()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 585 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 315 se::StreamExecutor* stream_exec, in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 100 GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) { in ExtractGpuExecutor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 118 GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) { in ExtractGpuExecutor()
|