| D | tiled_dot_emitter.cc | 658 int64 min_vectorization_width, int64 tile_size_m, in Config() argument
665 tile_size_m_(tile_size_m), in Config()
671 "_", min_vectorization_width(), "_", tile_size_m(), in GetCacheKey()
681 int64 tile_size_m() const { return tile_size_m_; } in tile_size_m() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter::Config
734 int64 tile_size_m, llvm::Value* m_start,
749 int64 tile_size_m() const { return config().tile_size_m(); } in tile_size_m() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter
823 const int64 m_end = dims().m() - dims().m() % tile_size_m(); in HandleResiduesOnM()
824 EmitTiledGemm(vsl, tile_size_k, k_start, k_end, n_start, n_end, tile_size_m(), in HandleResiduesOnM()
906 int64 tile_size_m, llvm::Value* m_start, llvm::Value* m_end) { in EmitTiledGemm() argument
907 ksl_.For("dot.m", m_start, m_end, tile_size_m, [&](llvm::Value* m_i) { in EmitTiledGemm()
[all …]
|