ifrt

type ConcreteEvenShardingProto struct {
	Devices    *DeviceListProto `protobuf:"bytes,1,opt,name=devices,proto3" json:"devices,omitempty"`
	MemoryKind *string          `protobuf:"bytes,4,opt,name=memory_kind,json=memoryKind,proto3,oneof" json:"memory_kind,omitempty"`
	Shape      *ShapeProto      `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	ShardShape *ShapeProto      `protobuf:"bytes,3,opt,name=shard_shape,json=shardShape,proto3" json:"shard_shape,omitempty"`
	// contains filtered or unexported fields
}

type ConcreteShardingProto struct {
	Devices     *DeviceListProto `protobuf:"bytes,1,opt,name=devices,proto3" json:"devices,omitempty"`
	MemoryKind  *string          `protobuf:"bytes,4,opt,name=memory_kind,json=memoryKind,proto3,oneof" json:"memory_kind,omitempty"`
	Shape       *ShapeProto      `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	ShardShapes []*ShapeProto    `protobuf:"bytes,3,rep,name=shard_shapes,json=shardShapes,proto3" json:"shard_shapes,omitempty"`
	// contains filtered or unexported fields
}

type DeviceListProto struct {

	// Serialization and deserialization are expected to ensure that device ids
	// are stable across proto construction and consumption.
	DeviceIds []int32 `protobuf:"varint,1,rep,packed,name=device_ids,json=deviceIds,proto3" json:"device_ids,omitempty"`
	// contains filtered or unexported fields
}

type OpaqueShardingProto struct {
	Devices    *DeviceListProto `protobuf:"bytes,1,opt,name=devices,proto3" json:"devices,omitempty"`
	MemoryKind *string          `protobuf:"bytes,2,opt,name=memory_kind,json=memoryKind,proto3,oneof" json:"memory_kind,omitempty"`
	// contains filtered or unexported fields
}

type Serialized struct {
	TypeName string `protobuf:"bytes,1,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"`
	Data     []byte `protobuf:"bytes,2,opt,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

type ShapeProto struct {
	Dims []int64 `protobuf:"varint,1,rep,packed,name=dims,proto3" json:"dims,omitempty"`
	// contains filtered or unexported fields
}

type SingleDeviceShardingProto struct {

	// Serialization and deserialization are expected to ensure that device ids
	// are stable across proto construction and consumption.
	DeviceId   int32   `protobuf:"varint,1,opt,name=device_id,json=deviceId,proto3" json:"device_id,omitempty"`
	MemoryKind *string `protobuf:"bytes,2,opt,name=memory_kind,json=memoryKind,proto3,oneof" json:"memory_kind,omitempty"`
	// contains filtered or unexported fields
}