Searched refs:to_shape (Results 1 – 7 of 7) 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.h | 654 const Shape& from_shape, const Shape& to_shape,
|
| D | shape_util.cc | 1202 const Shape& from_shape, const Shape& to_shape, in ReshapeLeavesDimensionsUnmodified() argument
1208 ShapeUtil::DimensionsUnmodifiedByReshape(from_shape, to_shape); in ReshapeLeavesDimensionsUnmodified()
|
| /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()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 202 TF_ASSIGN_OR_RETURN(auto to_shape, in ReshapeWithCorrectRepresentationAndSharding()
208 hlo_sharding, fast_mem, shape_representation_fn, &to_shape)); in ReshapeWithCorrectRepresentationAndSharding()
210 if (xla::ShapeUtil::Compatible(original_shape, to_shape)) { in ReshapeWithCorrectRepresentationAndSharding()
212 to_shape.set_dynamic_dimension(i, original_shape.is_dynamic_dimension(i)); in ReshapeWithCorrectRepresentationAndSharding()
215 return xla::Reshape(to_shape, original); in ReshapeWithCorrectRepresentationAndSharding()
|