Searched refs:bounded_shape (Results 1 – 5 of 5) sorted by relevance
168 const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible() argument169 if (dynamic_shape.rank() != bounded_shape.rank()) { in DynamicShapeIsCompatible()173 if (dynamic_shape.dimensions(i) > bounded_shape.dimensions(i)) { in DynamicShapeIsCompatible()
1576 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicArrayShapeIsCompatible() argument1577 if (dynamic_shape.rank() != bounded_shape.rank()) { in DynamicArrayShapeIsCompatible()1581 if (dynamic_shape.dimensions(i) > bounded_shape.dimensions(i)) { in DynamicArrayShapeIsCompatible()1589 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible() argument1594 auto subshape_result = TryGetSubshape(bounded_shape, index); in DynamicShapeIsCompatible()
698 const xla::Shape& bounded_shape);702 const xla::Shape& bounded_shape);
738 Literal LiteralBase::ToBoundedDynamic(const Shape& bounded_shape) const { in ToBoundedDynamic()739 CHECK(bounded_shape.is_dynamic()); in ToBoundedDynamic()740 Literal result(bounded_shape); in ToBoundedDynamic()
306 Literal ToBoundedDynamic(const Shape& bounded_shape) const;