Searched refs:space_dim (Results 1 – 2 of 2) sorted by relevance
1106 const int64 space_dim = result.second; in IsBroadcastPropagatable() local1108 return !absl::c_linear_search(broadcast_dims, space_dim); in IsBroadcastPropagatable()1136 const int64 space_dim = result.second; in SupportedOpForPropagation() local1138 << " space_dim " << space_dim << " reduce " in SupportedOpForPropagation()1141 absl::c_linear_search(reduce_dims, space_dim); in SupportedOpForPropagation()1267 const int64 space_dim = DimLookUp(permute_dims, old_space_dim); in Propagate() local1270 if (new_instr->shape().dimensions(space_dim) != in Propagate()1271 pivot_new_instr->shape().dimensions(space_dim)) { in Propagate()1274 CHECK_EQ(batch_dim + 1, space_dim); in Propagate()1280 new_dimensions[space_dim] *= (batch_size / old_batch_size); in Propagate()[all …]
1841 isl_size dim, space_dim, n_it; in isl_ast_build_product() local1851 space_dim = isl_space_dim(space, isl_dim_set); in isl_ast_build_product()1853 if (dim < 0 || space_dim < 0 || n_it < 0) in isl_ast_build_product()1855 dim += space_dim; in isl_ast_build_product()1870 strides = isl_vec_alloc(ctx, space_dim); in isl_ast_build_product()