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