| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 110 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
161 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
350 GpuVersion NVPTXCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
358 se::StreamExecutor* stream_exec, in CompileTargetBinary()
401 se::StreamExecutor* stream_exec, const string& ptx, in CompileGpuAsmOrGetCachedResult()
507 se::StreamExecutor* stream_exec, std::vector<std::vector<uint8>> modules) { in LinkModules()
|
| D | amdgpu_compiler.cc | 75 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
116 GpuVersion AMDGPUCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
131 se::StreamExecutor* stream_exec, in CompileTargetBinary()
|
| D | gpu_compiler.cc | 173 se::StreamExecutor* stream_exec) in GpuBfloat16Support()
241 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
537 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
631 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunHloPasses()
845 se::StreamExecutor* stream_exec, in CompileToTargetBinary()
1043 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
1147 GpuDeviceInfo GetGpuDeviceInfo(se::StreamExecutor* stream_exec) { in GetGpuDeviceInfo()
1291 absl::string_view entry_function_name, se::StreamExecutor* stream_exec, in CompileLmhloToExecutable()
|
| D | gemm_algorithm_picker.h | 32 GemmAlgorithmPicker(se::StreamExecutor* stream_exec, in GemmAlgorithmPicker()
|
| D | gpu_conv_algorithm_picker.h | 40 GpuConvAlgorithmPicker(se::StreamExecutor* stream_exec, in GpuConvAlgorithmPicker()
|
| D | gpu_compiler.h | 125 se::StreamExecutor* stream_exec, in LinkModules()
|
| D | stream_executor_util.cc | 312 tensorflow::mutex_lock LockGpu(const se::StreamExecutor* stream_exec) { in LockGpu()
331 absl::Span<const uint8> cubin_data, se::StreamExecutor* stream_exec) { in CreateKernel()
|
| D | gpu_conv_algorithm_picker.cc | 105 se::StreamExecutor* stream_exec) { in GetAlgorithms()
128 se::DeviceMemoryBase result_buffer, se::StreamExecutor* stream_exec, in GetMIOpenAlgorithms()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator_test.cc | 49 auto stream_exec = ExecutorForPlatformDeviceId(platform_device_id); in TEST() local
145 auto stream_exec = ExecutorForPlatformDeviceId(platform_device_id); in TEST() local
187 auto stream_exec = ExecutorForPlatformDeviceId(platform_device_id); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 45 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
51 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
62 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable, in CompileTargetBinary()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 106 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
131 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/cuda/ |
| D | redzone_allocator_test.cc | 55 StreamExecutor* stream_exec = platform->ExecutorForDevice(0).ValueOrDie(); in TEST() local
130 StreamExecutor* stream_exec = platform->ExecutorForDevice(0).ValueOrDie(); in TEST() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | event.cc | 24 Event::Event(StreamExecutor* stream_exec) in Event()
|
| D | stream_executor_pimpl.cc | 75 ScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in ScopedTracer()
116 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()
|
| D | gpu_executor.h | 404 inline GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) { in ExtractGpuExecutor()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 29 explicit DeviceHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in DeviceHostAllocator()
|
| D | device_mem_allocator.h | 31 explicit DeviceMemAllocator(se::StreamExecutor* stream_exec, in DeviceMemAllocator()
|
| D | device_event_mgr_test.cc | 112 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
121 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
440 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in BM_no_ops() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 134 se::StreamExecutor* stream_exec, in LogConvAutotuneResults()
178 se::StreamExecutor* stream_exec, absl::Span<const AutotuneResult> results) { in LogFusedConvForwardAutotuneResults()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 54 se::StreamExecutor* stream_exec = stream->parent(); in CompileMlirModule() local
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.cc | 59 auto stream_exec = stream->parent(); in AllocateOutputMemoryWithInputReuse() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 232 std::vector<std::vector<se::StreamExecutor*>> stream_exec, in Compile()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 112 std::vector<std::vector<stream_executor::StreamExecutor*>> stream_exec, in Compile()
|