Searched refs:n_i (Results 1 – 2 of 2) sorted by relevance
/external/llvm-project/polly/lib/External/isl/ |
D | isl_schedule_constraints.c | 753 isl_size n_i; in isl_schedule_constraints_n_map() local 755 n_i = isl_union_map_n_map(sc->constraint[i]); in isl_schedule_constraints_n_map() 756 if (n_i < 0) in isl_schedule_constraints_n_map() 758 n += n_i; in isl_schedule_constraints_n_map()
|
/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()
|