Searched refs:to_shape (Results 1 – 4 of 4) sorted by relevance
32 std::function<bool(const Shape& from_shape, const Shape& to_shape)>;39 bool ReshapeIsBitcast(const Shape& from_shape, const Shape& to_shape) const { in ReshapeIsBitcast() argument44 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape); in ReshapeIsBitcast()46 return reshape_is_bitcast_callback_(from_shape, to_shape); in ReshapeIsBitcast()
86 const Shape& to_shape) { in ConvertTupleElements() argument94 const Shape& to_ele_shape = ShapeUtil::GetTupleElementShape(to_shape, i); in ConvertTupleElements()
35 Status ShapeLayout::AssignLayoutToShape(Shape* to_shape) const { in AssignLayoutToShape()36 if (!ShapeUtil::Compatible(*to_shape, shape_)) { in AssignLayoutToShape()38 ShapeUtil::HumanString(*to_shape), in AssignLayoutToShape()41 *to_shape = shape_; in AssignLayoutToShape()
43 Status AssignLayoutToShape(Shape* to_shape) const;