Home
last modified time | relevance | path

Searched defs:stream_exec (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.cc70 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
87 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
119 GpuVersion AMDGPUCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
135 se::StreamExecutor* stream_exec) { in CompileTargetBinary()
Dnvptx_compiler.cc113 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
155 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
326 GpuVersion NVPTXCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
343 se::StreamExecutor* stream_exec) { in CompileTargetBinary()
389 se::StreamExecutor* stream_exec, const string& ptx, int cc_major, in CompileGpuAsmOrGetCachedResult()
Dgemm_algorithm_picker.h32 GemmAlgorithmPicker(se::StreamExecutor* stream_exec, in GemmAlgorithmPicker()
Dgpu_compiler.cc120 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
318 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunHloPasses()
334 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
Dgpu_conv_algorithm_picker.h40 GpuConvAlgorithmPicker(se::StreamExecutor* stream_exec, in GpuConvAlgorithmPicker()
Dstream_executor_util.cc176 tensorflow::mutex_lock LockGpu(const se::StreamExecutor* stream_exec) { in LockGpu()
195 absl::Span<const uint8> cubin_data, se::StreamExecutor* stream_exec) { in CreateKernel()
Dgpu_conv_algorithm_picker.cc100 se::StreamExecutor* stream_exec) { in GetAlgorithms()
124 se::DeviceMemoryBase result_buffer, se::StreamExecutor* stream_exec, in GetAlgorithms()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc45 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
51 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
56 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) { return 0; } in GetGpuVersion()
60 GpuVersion gpu_version, se::StreamExecutor* stream_exec) { in CompileTargetBinary()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator_test.cc45 auto stream_exec = in TEST() local
145 auto stream_exec = in TEST() local
192 auto stream_exec = in TEST() local
Dgpu_event_mgr_test.cc117 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
134 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
167 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
184 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
203 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
225 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
247 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
266 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST() local
589 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in BM_no_ops() local
Dgpu_host_allocator.h29 explicit GpuHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in GpuHostAllocator()
Dgpu_mem_allocator.h31 explicit GPUMemAllocator(se::StreamExecutor* stream_exec, in GPUMemAllocator()
/external/tensorflow/tensorflow/compiler/xla/
Ddevice_util.h32 string DeviceIdentifier(se::StreamExecutor* stream_exec) { in DeviceIdentifier()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dfailover_compiler.cc30 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunHloPasses()
42 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
Dmlir_compiler.cc142 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunHloPasses()
176 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
457 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc101 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
126 std::vector<std::vector<se::StreamExecutor*>> stream_exec, in Compile()
/external/tensorflow/tensorflow/stream_executor/
Devent.cc24 Event::Event(StreamExecutor* stream_exec) in Event()
Dstream_executor_pimpl.cc74 ScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in ScopedTracer()
115 MakeScopedTracer(StreamExecutor *stream_exec, BeginCallT begin_call, in MakeScopedTracer()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_activation.cc34 StreamExecutor* stream_exec) in ScopedActivateExecutorContext()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dredzone_allocator_test.cc55 StreamExecutor* stream_exec = platform->ExecutorForDevice(0).ValueOrDie(); in TEST() local
130 StreamExecutor* stream_exec = platform->ExecutorForDevice(0).ValueOrDie(); in TEST() local
Dcuda_gpu_executor.cc118 GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) { in ExtractGpuExecutor()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h34 inline bool IsVoltaOrLater(const se::StreamExecutor& stream_exec) { in IsVoltaOrLater()
164 se::StreamExecutor* stream_exec) const { in ShouldIncludeWinogradNonfusedAlgo()
Dgpu_utils.cc135 se::StreamExecutor* stream_exec, in LogConvAutotuneResults()
178 se::StreamExecutor* stream_exec, absl::Span<const AutotuneResult> results) { in LogFusedConvForwardAutotuneResults()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc560 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, in RunBackend()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc99 GpuExecutor* ExtractGpuExecutor(StreamExecutor* stream_exec) { in ExtractGpuExecutor()