Searched refs:from_shape (Results 1 – 4 of 4) sorted by relevance
| /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()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| 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/client/ |
| D | xla_builder.cc | 579 const Shape& from_shape = should_broadcast_lhs ? *lhs_shape : *rhs_shape; in BinaryOp() local
587 for (int64 from_dim = 0; from_dim < from_shape.rank(); from_dim++) { in BinaryOp()
589 to_size[to_dim] = from_shape.dimensions(from_dim); in BinaryOp()
590 to_size_is_dynamic[to_dim] = from_shape.is_dynamic_dimension(from_dim); in BinaryOp()
594 from_shape.element_type(), to_size, to_size_is_dynamic); in BinaryOp()
|