Классы
| BroadcastHelper <T расширяет TType > | Вспомогательный оператор для проведения трансляций в стиле XLA. Транслирует `lhs` и `rhs` в один и тот же ранг, добавляя измерения размера 1 к любому из `lhs` и `rhs`, имеющему более низкий ранг, используя правила трансляции XLA для бинарных операторов. |
| ClusterOutput <T расширяет TType > | Оператор, который соединяет выходные данные вычисления XLA с другими узлами потребительского графа. |
| Conv <T расширяет TType > | Обертывает оператор XLA ConvGeneralDilated, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#conv_convolution. |
| Деквантовать | Берет упакованный ввод uint32 и распаковывает его в uint8, чтобы сделать Деквантование на устройстве. |
| Точка <T расширяет TType > | Обертывает оператор XLA DotGeneral, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#dotgeneral. |
| DynamicSlice <T расширяет TType > | Обертывает оператор XLA DynamicSlice, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#dynamicslice. |
| DynamicUpdateSlice <T расширяет TType > | Обертывает оператор XLA DynamicUpdateSlice, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#dynamicupdateslice. |
| Einsum <T расширяет TType > | Операция, поддерживающая базовую операцию einsum с 2 входами и 1 выходом. |
| Соберите <T расширяет TType > | Обертывает оператор XLA Gather, описанный по адресу https://www.tensorflow.org/xla/operation_semantics#gather |
| KeyValueSort <T расширяет TNumber , U расширяет TType > | Обертывает оператор сортировки XLA, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#sort. |
| Pad <T расширяет TType > | Обертывает оператор XLA Pad, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#pad. |
| Recv <T расширяет TType > | Получает именованный тензор из другого вычисления XLA. |
| Репликаид | Идентификатор реплики. |
| SelfAdjointEig <T расширяет TType > | Вычисляет собственное разложение пакета самосопряженных матриц. (Примечание: поддерживаются только реальные входные данные). |
| Отправлять | Отправляет именованный тензор на другое вычисление XLA. |
| Шардинг <T расширяет TType > | Операция, которая сегментирует входные данные на основе заданного атрибута сегментирования. |
| Сортировка <T расширяет TType > | Обертывает оператор сортировки XLA, описанный по адресу https://www.tensorflow.org/ Performance/xla/operation_semantics#sort. |
| Svd <T расширяет TType > | Вычисляет собственное разложение пакета самосопряженных матриц. (Примечание: поддерживаются только реальные входные данные). |
| XlaRecvFromHost <T расширяет TType > | Операция по получению тензора от хоста. |
| КслаСендтоХост | Операция по отправке тензора на хост. |
| XlaSetBound | Установите границу для данного входного значения в качестве подсказки для компилятора Xla, возвращает то же значение. |