Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc615 class TiledSmallGemmEmitter { class
696 explicit TiledSmallGemmEmitter(Config config, llvm::Value* lhs, in TiledSmallGemmEmitter() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter
762 void TiledSmallGemmEmitter::Emit() { HandleResiduesOnN(); } in Emit()
764 void TiledSmallGemmEmitter::HandleResiduesOnN() { in HandleResiduesOnN()
803 void TiledSmallGemmEmitter::HandleResiduesOnK(VectorSupportLibrary* vsl, in HandleResiduesOnK()
820 void TiledSmallGemmEmitter::HandleResiduesOnM( in HandleResiduesOnM()
903 void TiledSmallGemmEmitter::EmitTiledGemm( in EmitTiledGemm()
1050 TiledSmallGemmEmitter::Config config( in EmitSmallGemm()
1052 TiledSmallGemmEmitter::Dimensions{/*m=*/m, /*k=*/k, /*n=*/n}, in EmitSmallGemm()
1061 TiledSmallGemmEmitter small_gemm_emitter(config, /*lhs=*/lhs, in EmitSmallGemm()