Searched defs:inferred_dimension (Results 1 – 7 of 7) sorted by relevance
365 int64 inferred_dimension) { in ReshapeInternal()
740 int64 inferred_dimension() const { return inferred_dimension_; } in inferred_dimension() function
743 int64 inferred_dimension = -1; in CreateFromProto() local1447 const Shape& shape, HloInstruction* operand, int64 inferred_dimension) { in CreateReshape()3921 int64 HloInstruction::inferred_dimension() const { in inferred_dimension() function in xla::HloInstruction
2945 absl::Span<const int64> new_sizes, int64 inferred_dimension) { in InferReshapeShape()
1361 optional<int64> inferred_dimension; in ParseInstructionRhs() local
1102 int64 inferred_dimension) in HloReshapeInstruction()
941 int64 inferred_dimension) { in ReshapeInternal()1117 int64 inferred_dimension) { in Reshape()1131 int64 inferred_dimension) { in Reshape()1141 int64 inferred_dimension) { in Reshape()4146 int64 inferred_dimension) { in ReshapeWithInferredDimension()