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.cc160 Status NVPTXCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment() function in xla::gpu::NVPTXCompiler
172 TF_RETURN_IF_ERROR(GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()
Dnvptx_compiler.h45 Status OptimizeHloPostLayoutAssignment(
Dgpu_compiler.h74 virtual Status OptimizeHloPostLayoutAssignment(
Dgpu_compiler.cc306 TF_RETURN_IF_ERROR(OptimizeHloPostLayoutAssignment(hlo_module, stream_exec, in OptimizeHloModule()
402 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