Searched refs:dim_idx (Results 1 – 6 of 6) 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 …]
635 int dim_idx = 4;639 output_tensor->flat<Tout>()(dim_idx++) =642 output_tensor->flat<Tout>()(dim_idx++) = 0.0;839 int dim_idx = 4;841 static_output[dim_idx++] = static_cast<Tout>(tensor.dim_size(i));
140 for (int64 dim_idx = 0; dim_idx < padded->shape().dimensions_size(); in RewriteReduction() local141 dim_idx++) { in RewriteReduction()142 if (dim_idx == reduced_input_dimension) { in RewriteReduction()146 reshaped_dimensions.push_back(padded->shape().dimensions(dim_idx)); in RewriteReduction()
173 for (int dim_idx = 0; dim_idx < ndims_input; ++dim_idx) in ExpandInputDimsToOutputShape() local174 reshaped_dims->at(dim_idx + dim_offset) = input_dims[dim_idx]; in ExpandInputDimsToOutputShape()
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++;
400 Tensor dim_idx(DT_INT32, TensorShape({})); in SparsifyGatherInternal() local401 dim_idx.flat<int32>()(0) = -1; in SparsifyGatherInternal()406 SetNodeAttr("value", dim_idx, &dim_idx_node); in SparsifyGatherInternal()