Searched defs:computation_device (Results 1 – 2 of 2) sorted by relevance
92 DeviceAssignmentProto::ComputationDevice* computation_device = in Serialize() local114 const auto& computation_device = proto.computation_devices(computation); in Deserialize() local
127 auto* computation_device = device_assignment.add_computation_devices(); in UpdateTPUCompileMetadata() local