Home
last modified time | relevance | path

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

/external/llvm-project/polly/lib/External/isl/
Disl_schedule_constraints.c753 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/
Dtiled_dot_emitter.cc796 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()