Home
last modified time | relevance | path

Searched refs:OptimizeHloPostLayoutAssignment (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Damdgpu_compiler.cc108 Status AMDGPUCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::AMDGPUCompiler
111 TF_RETURN_IF_ERROR(GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()
H A Dnvptx_compiler.cc119 Status NVPTXCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::NVPTXCompiler
146 TF_RETURN_IF_ERROR(GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()
H A Damdgpu_compiler.h40 Status OptimizeHloPostLayoutAssignment(
H A Dnvptx_compiler.h44 Status OptimizeHloPostLayoutAssignment(
H A Dgpu_compiler.h121 virtual Status OptimizeHloPostLayoutAssignment(
H A Dgpu_compiler.cc605 TF_RETURN_IF_ERROR(OptimizeHloPostLayoutAssignment(hlo_module, stream_exec, in OptimizeHloModule()
722 Status GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::GpuCompiler
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dllvm_compiler_test.cc52 Status OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::GpuDummyCompiler