Searched refs:reshaped_dimensions (Results 1 – 2 of 2) sorted by relevance
1136 std::vector<int64> reshaped_dimensions; in CreateBroadcastSequence() local1140 reshaped_dimensions.push_back(operand->shape().dimensions(i)); in CreateBroadcastSequence()1151 reshaped_dimensions), in CreateBroadcastSequence()
449 std::vector<int64> reshaped_dimensions; in AddBroadcastSequence() local453 reshaped_dimensions.push_back(operand_shape.dimensions(i)); in AddBroadcastSequence()464 reshaped_dimensions), in AddBroadcastSequence()