Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment.cc87 if (!(IsCublasGemm(hlo) || IsMatrixMultiplication(hlo))) { in ComputeStreamToAssign()
155 if (IsCublasGemm(*hlo) || IsMatrixMultiplication(*hlo)) { in AssignStreams()
Dir_emission_utils.h75 bool IsCublasGemm(const HloInstruction& hlo);
Dgemm_algorithm_picker.cc307 if (IsCublasGemm(*instr)) { in RunOnComputation()
Dgpu_layout_assignment.cc209 CHECK(!IsCublasGemm(*instruction)) in AddBackendConstraints()
Dnvptx_compiler.cc189 if (IsCublasGemm(*user)) { in CanShareBufferHint()
Dir_emission_utils.cc128 bool IsCublasGemm(const HloInstruction& hlo) { in IsCublasGemm() function
212 return IsCublasGemm(hlo) || IsCustomCallToDnnBatchNorm(hlo) || in ImplementedAsLibraryCall()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc623 if (xla::gpu::IsCublasGemm(*instr)) { in EmitCustomCallOp()