Searched refs:to_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | layout_util.cc | 91 auto to_shape, in ReshapeWithCorrectRepresentationAndSharding()
101 &to_shape)); in ReshapeWithCorrectRepresentationAndSharding()
103 if (xla::ShapeUtil::Compatible(original_shape, to_shape)) { in ReshapeWithCorrectRepresentationAndSharding()
105 to_shape.set_dynamic_dimension(i, original_shape.is_dynamic_dimension(i)); in ReshapeWithCorrectRepresentationAndSharding()
108 return xla::Reshape(to_shape, original); in ReshapeWithCorrectRepresentationAndSharding()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | layout_util.cc | 106 TF_ASSIGN_OR_RETURN(auto to_shape, in ReshapeWithCorrectRepresentationAndSharding()
114 hlo_sharding, fast_mem, shape_determination_fns, &to_shape)); in ReshapeWithCorrectRepresentationAndSharding()
116 if (xla::ShapeUtil::Compatible(original_shape, to_shape)) { in ReshapeWithCorrectRepresentationAndSharding()
118 to_shape.set_dynamic_dimension(i, original_shape.is_dynamic_dimension(i)); in ReshapeWithCorrectRepresentationAndSharding()
121 return xla::Reshape(to_shape, original); in ReshapeWithCorrectRepresentationAndSharding()
|
| /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 | 563 const Shape& from_shape, const Shape& to_shape,
|
| D | shape_util.cc | 1278 const Shape& from_shape, const Shape& to_shape, in ReshapeLeavesDimensionsUnmodified() argument
1286 ShapeUtil::DimensionsUnmodifiedByReshape(from_shape, to_shape); in ReshapeLeavesDimensionsUnmodified()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bitcast_dtypes_expander.cc | 42 const Shape& to_shape = instruction->shape(); in ExpandInstruction() local
45 int output_bit_width = primitive_util::BitWidth(to_shape.element_type()); in ExpandInstruction()
58 to_shape.ToString()); in ExpandInstruction()
104 BitcastConvertType(input, to_shape.element_type()); in ExpandInstruction()
|
| D | algebraic_simplifier.h | 37 std::function<bool(const Shape& from_shape, const Shape& to_shape)>;
50 bool ReshapeIsBitcast(const Shape& from_shape, const Shape& to_shape) const { in ReshapeIsBitcast() argument
55 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape); in ReshapeIsBitcast()
57 return reshape_is_bitcast_callback_(from_shape, to_shape); in ReshapeIsBitcast()
|
| D | hlo_element_type_converter.cc | 88 const Shape& to_shape) { in ConvertTupleElements() argument
96 const Shape& to_ele_shape = ShapeUtil::GetTupleElementShape(to_shape, i); in ConvertTupleElements()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_symbolic_shape_optimization.cc | 185 ShapeComponentAnalysis& analysis, Value from, Value to_shape) { in isNonExpandingBroadcast() argument
187 auto out_shape = analysis.GetValueInfo(to_shape); in isNonExpandingBroadcast()
|