Searched refs:ShardedDeviceArrayObject (Results 1 – 1 of 1) sorted by relevance
40 struct ShardedDeviceArrayObject { struct47 static_assert(std::is_standard_layout<ShardedDeviceArrayObject>::value,52 ShardedDeviceArrayObject* self = reinterpret_cast<ShardedDeviceArrayObject*>( in sharded_device_array_tp_new()61 ShardedDeviceArrayObject* o = in sharded_device_array_tp_dealloc()62 reinterpret_cast<ShardedDeviceArrayObject*>(self); in sharded_device_array_tp_dealloc()126 ShardedDeviceArrayObject* sda = in Make()127 reinterpret_cast<ShardedDeviceArrayObject*>(obj.ptr()); in Make()140 return &(reinterpret_cast<ShardedDeviceArrayObject*>(handle.ptr())->sda); in AsShardedDeviceArrayUnchecked()153 offsetof(ShardedDeviceArrayObject, sda)); in AsHandle()201 type->tp_basicsize = sizeof(ShardedDeviceArrayObject); in RegisterTypes()[all …]