Searched refs:n_i (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 796 ksl_.For("epi.n", n_start, dims().n(), 1, [&](llvm::Value* n_i) { in HandleResiduesOnN() argument
797 llvm::Value* n_i_next = b_->CreateAdd(n_i, b_->getInt64(1)); in HandleResiduesOnN()
798 HandleResiduesOnK(&vsl, n_i, n_i_next); in HandleResiduesOnN()
917 "dot.n", n_start, n_end, vsl->vector_size(), [&](llvm::Value* n_i) { in EmitTiledGemm() argument
918 TileVariable result_tile_var(vsl, result_memory_tile.LoadTile(n_i)); in EmitTiledGemm()
924 std::vector<llvm::Value*> rhs_tile = rhs_memory_tile.LoadTile(n_i); in EmitTiledGemm()
936 result_memory_tile.StoreTile(result_tile_var.Get(), n_i); in EmitTiledGemm()
|