Searched defs:inferred_dimension (Results 1 – 7 of 7) sorted by relevance
486 int64_t inferred_dimension) { in ReshapeInternal()
922 int64_t inferred_dimension = -1; in CreateFromProto() local1746 const Shape& shape, HloInstruction* operand, int64_t inferred_dimension) { in CreateReshape()4489 int64_t HloInstruction::inferred_dimension() const { in inferred_dimension() function in xla::HloInstruction
991 int64_t inferred_dimension() const { return inferred_dimension_; } in inferred_dimension() function
3085 absl::Span<const int64_t> new_sizes, int64_t inferred_dimension) { in InferReshapeShape()
1737 optional<int64_t> inferred_dimension; in CreateInstruction() local
1291 int64_t inferred_dimension) in HloReshapeInstruction()
1007 int64_t inferred_dimension) { in ReshapeInternal()1184 int64_t inferred_dimension) { in Reshape()1198 int64_t inferred_dimension) { in Reshape()1208 int64_t inferred_dimension) { in Reshape()4102 int64_t inferred_dimension) { in ReshapeWithInferredDimension()