Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.h48 Status OptimizeHloPostLayoutAssignment(
Dnvptx_compiler.cc160 Status NVPTXCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::NVPTXCompiler
184 TF_RETURN_IF_ERROR(GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()
Dgpu_compiler.h69 virtual Status OptimizeHloPostLayoutAssignment(
Dgpu_compiler.cc441 TF_RETURN_IF_ERROR(OptimizeHloPostLayoutAssignment(hlo_module, stream_exec, in OptimizeHloModule()
536 Status GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::GpuCompiler
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc50 Status OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::GpuDummyCompiler