Searched refs:ellipsis_axis (Results 1 – 1 of 1) sorted by relevance
185 int ellipsis_axis, Labels* labels, in InsertBroadcastLabels()187 labels->erase(labels->begin() + ellipsis_axis); in InsertBroadcastLabels()188 labels->insert(labels->begin() + ellipsis_axis, num_bcast_dims, 0); in InsertBroadcastLabels()189 std::iota(labels->begin() + ellipsis_axis, in InsertBroadcastLabels()190 labels->begin() + ellipsis_axis + num_bcast_dims, in InsertBroadcastLabels()259 int ellipsis_axis = -1; in ProcessDimensions() local264 ellipsis_axis = label_idx; in ProcessDimensions()269 label_idx + (ellipsis_axis == -1 ? 0 : num_bcast_dims - 1); in ProcessDimensions()275 if (ellipsis_axis != -1) { in ProcessDimensions()276 InsertBroadcastLabels(num_bcast_dims, num_named_labels, ellipsis_axis, in ProcessDimensions()[all …]