Операция, используемая средством разделения XLA SPMD для переключения с автоматического разделения на
ручное разделение. Он аннотирует входные данные (полные, для автоматического секционирования) с тем же сегментированием, которое используется при ручном секционировании, и выводит тензор в форме осколка, который будет использоваться последующими операциями, секционируемыми вручную. Если фигура не разделена равномерно, область заполнения будет замаскирована нулями.
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Унаследованные методы
логическое значение | равно (Объект arg0) |
последний класс<?> | получитьКласс () |
интервал | хэш-код () |
окончательная пустота | поставить в известность () |
окончательная пустота | уведомитьВсе () |
Нить | нанизывать () |
окончательная пустота | подождать (длинный arg0, int arg1) |
окончательная пустота | подождите (длинный arg0) |
окончательная пустота | ждать () |
абстрактная среда выполнения | окружение () Верните среду выполнения, в которой была создана эта операция. |
абстрактная операция |
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static XlaSpmdFullToShardShape <T> create (область действия , ввод операнда <T>, строка manualSharding)
Фабричный метод для создания класса, обертывающего новую операцию XlaSpmdFullToShardShape.
Параметры
объем | текущий объем |
---|
Возврат
- новый экземпляр XlaSpmdFullToShardShape