Searched refs:from_shape (Results 1 – 3 of 3) 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()
355 def _shape(batch_size, from_shape): argument356 if (not isinstance(from_shape, tensor_shape.TensorShape) or357 from_shape.ndims == 0):363 return tensor_shape.TensorShape([batch_size]).concatenate(from_shape)
514 const Shape& from_shape = should_broadcast_lhs ? lhs_shape : rhs_shape; in BinaryOp() local522 for (int64 from_dim = 0; from_dim < from_shape.rank(); from_dim++) { in BinaryOp()524 to_size[to_dim] = from_shape.dimensions(from_dim); in BinaryOp()525 to_size_is_dynamic[to_dim] = from_shape.is_dynamic_dimension(from_dim); in BinaryOp()529 from_shape.element_type(), to_size, to_size_is_dynamic); in BinaryOp()