Searched defs:inferred_dimension (Results 1 – 6 of 6) sorted by relevance
631 int64 inferred_dimension() const { return inferred_dimension_; } in inferred_dimension() function
638 int64 inferred_dimension = -1; in CreateFromProto() local1281 const Shape& shape, HloInstruction* operand, int64 inferred_dimension) { in CreateReshape()3571 int64 HloInstruction::inferred_dimension() const { in inferred_dimension() function in xla::HloInstruction
2802 absl::Span<const int64> new_sizes, int64 inferred_dimension) { in InferReshapeShape()
941 optional<int64> inferred_dimension; in ParseInstructionRhs() local
963 int64 inferred_dimension) in HloReshapeInstruction()
831 int64 inferred_dimension) { in Reshape()1013 int64 inferred_dimension) { in Reshape()1027 int64 inferred_dimension) { in Reshape()2925 int64 inferred_dimension) { in ReshapeWithInferredDimension()