Home
last modified time | relevance | path

Searched refs:DistributedVariable (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Ddistributed.py24 class DistributedVariable(VariableTracker): class
37 if not DistributedVariable.is_available():
51 if not DistributedVariable.is_available():
59 if not DistributedVariable.is_available():
81 class WorldMetaClassVariable(DistributedVariable):
104 class PlacementClassVariable(DistributedVariable):
108 if not DistributedVariable.is_available():
139 class PlacementVariable(DistributedVariable):
143 if not DistributedVariable.is_available():
203 class DeviceMeshVariable(DistributedVariable):
[all …]
H A D__init__.py30 from .distributed import BackwardHookVariable, DistributedVariable, PlacementVariable
H A Dtorch.py46 from .distributed import DistributedVariable, ProcessGroupVariable
623 if DistributedVariable.is_available():
H A Dbuiltin.py1721 variables.DistributedVariable,
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_utils.py100 if same_id and isinstance(v0, values_lib.DistributedVariable):
139 if (isinstance(x, values_lib.DistributedVariable) or
214 not isinstance(val, values_lib.DistributedVariable)):
241 if not isinstance(v, values_lib.DistributedVariable):
363 if isinstance(val, values_lib.DistributedVariable):
370 if isinstance(val, values_lib.DistributedVariable):
451 "VariableClass": values_lib.DistributedVariable,
H A Dvalues_v2.py33 class DistributedVariable(resource_variable_ops.BaseResourceVariable): class
252 return DistributedVariable(
266 ops.register_tensor_conversion_function(DistributedVariable, _tensor_conversion)
H A Dmirrored_strategy.py529 assert not isinstance(v, values.DistributedVariable)
772 assert isinstance(var, values.DistributedVariable)
819 if isinstance(var, values.DistributedVariable):
H A Dvalues.py477 class DistributedVariable(DistributedDelegate, variables_lib.Variable, class
490 super(DistributedVariable, self).__init__(values)
1094 class MirroredVariable(DistributedVariable, Mirrored):
1210 class SyncOnReadVariable(DistributedVariable):
1405 ops.register_tensor_conversion_function(DistributedVariable,
H A Dvalues_v2_test.py370 return values_v2.DistributedVariable(variables)
394 return values_v2.DistributedVariable(
H A Dtpu_strategy.py142 return flat and isinstance(flat[0], values.DistributedVariable)
1296 values.DistributedVariable) and value._packed_variable is not None:
1361 values.DistributedVariable) and value._packed_variable is not None:
H A Dtf_function_test.py125 if isinstance(v, values.DistributedVariable):
H A Dtpu_values.py153 class TPUDistributedVariable(TPUVariableMixin, values.DistributedVariable):
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/
H A Dsaved_model_test.py672 self.assertIsInstance(loaded.v1, values.DistributedVariable)