Searched refs:inferred_dimension_ (Results 1 – 2 of 2) sorted by relevance
740 int64 inferred_dimension() const { return inferred_dimension_; } in inferred_dimension()754 int64 inferred_dimension_; variable
1104 inferred_dimension_(inferred_dimension) { in HloReshapeInstruction()1110 if (inferred_dimension_ != -1) { in ToProto()1111 proto.add_dimensions(inferred_dimension_); in ToProto()