Searched refs:dev_assign_proto (Results 1 – 3 of 3) sorted by relevance
66 xla::DeviceAssignmentProto dev_assign_proto; in LoadProgramAndEnqueueToStream() local67 TF_RETURN_IF_ERROR(dev_assign->Serialize(&dev_assign_proto)); in LoadProgramAndEnqueueToStream()69 stream_executor::tpu::SerializeProto(dev_assign_proto); in LoadProgramAndEnqueueToStream()
40 xla::DeviceAssignmentProto dev_assign_proto; in ToC() local43 ->Serialize(&dev_assign_proto) in ToC()46 stream_executor::tpu::SerializeProto(dev_assign_proto); in ToC()