Searched refs:to_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_layout.cc | 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()
|
| D | shape_layout.h | 43 Status AssignLayoutToShape(Shape* to_shape) const;
|
| D | shape_util.cc | 1073 const Shape& from_shape, const Shape& to_shape, in ReshapeLeavesDimensionsUnmodified() argument
1079 ShapeUtil::DimensionsUnmodifiedByReshape(from_shape, to_shape); in ReshapeLeavesDimensionsUnmodified()
|
| D | shape_util.h | 605 const Shape& from_shape, const Shape& to_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.h | 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() argument
44 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape); in ReshapeIsBitcast()
46 return reshape_is_bitcast_callback_(from_shape, to_shape); in ReshapeIsBitcast()
|
| D | hlo_element_type_converter.cc | 86 const Shape& to_shape) { in ConvertTupleElements() argument
94 const Shape& to_ele_shape = ShapeUtil::GetTupleElementShape(to_shape, i); in ConvertTupleElements()
|