Searched refs:XLAOptions (Results 1 – 4 of 4) sorted by relevance
1 path: "tensorflow.tpu.XLAOptions"3 is_instance: "<class \'tensorflow.python.tpu.tpu.XLAOptions\'>"4 is_instance: "<class \'tensorflow.python.tpu.tpu.XLAOptions\'>"
12 name: "XLAOptions"
864 class XLAOptions( class877 return super(XLAOptions, cls).__new__(cls, use_spmd_for_xla_partitioning)889 xla_options: Optional[XLAOptions] = None) -> List[Any]: argument1183 xla_options: Optional[XLAOptions] = None, argument1243 xla_options = xla_options or XLAOptions()1719 xla_options: Optional[XLAOptions] = None, argument1876 xla_options: Optional[XLAOptions] = None) -> List[core_types.Tensor]: argument1959 xla_options: Optional[XLAOptions] = None): argument2020 xla_options: Optional[XLAOptions] = None) -> Any: argument
1033 xla_options=tpu.XLAOptions(use_spmd_for_xla_partitioning=self1489 xla_options=tpu.XLAOptions(use_spmd_for_xla_partitioning=self