Searched refs:dim_idx (Results 1 – 3 of 3) sorted by relevance
95 int dim_idx = 0; in ReduceTransposeDimensions() local101 combined_dims[dim_idx] *= shape.dim_size(cur_head); in ReduceTransposeDimensions()105 dim_idx++; in ReduceTransposeDimensions()106 new_dim_position[cur_head] = dim_idx; in ReduceTransposeDimensions()107 combined_dims[dim_idx] = shape.dim_size(cur_head); in ReduceTransposeDimensions()111 new_perm->resize(dim_idx + 1); in ReduceTransposeDimensions()112 new_dims->resize(dim_idx + 1); in ReduceTransposeDimensions()113 dim_idx = 0; in ReduceTransposeDimensions()117 (*new_perm)[dim_idx] = new_perm_idx; in ReduceTransposeDimensions()118 (*new_dims)[dim_idx] = combined_dims[new_perm_idx]; in ReduceTransposeDimensions()[all …]
235 int dim_idx = 0;249 m_dimensions[dim_idx] = eval_left_dims[i];251 if (dim_idx != i) {260 dim_idx++;276 m_dimensions[dim_idx] = eval_right_dims[i];284 dim_idx++;
401 Tensor dim_idx(DT_INT32, TensorShape({})); in SparsifyGatherInternal() local402 dim_idx.flat<int32>()(0) = -1; in SparsifyGatherInternal()407 SetNodeAttr("value", dim_idx, &dim_idx_node); in SparsifyGatherInternal()