Searched refs:sharded_shape (Results 1 – 1 of 1) sorted by relevance
62 Shape sharded_shape; in GetShardedShape() local64 TF_ASSIGN_OR_RETURN(sharded_shape, in GetShardedShape()67 sharded_shape = unsharded_shape; in GetShardedShape()69 LayoutUtil::ClearLayout(&sharded_shape); in GetShardedShape()70 return sharded_shape; in GetShardedShape()171 const Shape& sharded_shape, Shape* dst_shape) { in DetermineArgumentLayoutsFromCompileOptions() argument175 CHECK(ShapeUtil::IndexIsValid(sharded_shape, idx)); in DetermineArgumentLayoutsFromCompileOptions()177 ShapeUtil::GetSubshape(sharded_shape, idx); in DetermineArgumentLayoutsFromCompileOptions()