Searched refs:identity_dims (Results 1 – 1 of 1) sorted by relevance
874 std::vector<int64> identity_dims(lhs.rank()); in InferElementwiseBinaryOpShape() local875 std::iota(identity_dims.begin(), identity_dims.end(), 0); in InferElementwiseBinaryOpShape()877 broadcast_dimensions != identity_dims) { in InferElementwiseBinaryOpShape()