D | tiled_dot_emitter.cc | 737 llvm::Value* GetInt64(int64 value) { return b_->getInt64(value); } in GetInt64() function in xla::cpu::__anon016ac5180111::TiledSmallGemmEmitter 782 HandleResiduesOnK(&vsl, GetInt64(n_start), GetInt64(n_end)); in HandleResiduesOnN() 809 HandleResiduesOnM(vsl, tile_size_k(), GetInt64(k_start), GetInt64(k_end), in HandleResiduesOnK() 815 HandleResiduesOnM(vsl, dims().k() - k_start, GetInt64(k_start), in HandleResiduesOnK() 816 GetInt64(dims().k()), n_start, n_end); in HandleResiduesOnK() 825 GetInt64(0), GetInt64(m_end)); in HandleResiduesOnM() 829 dims().m() - m_end, GetInt64(m_end), GetInt64(dims().m())); in HandleResiduesOnM()
|