Searched refs:PackedVarAndDevice (Results 1 – 4 of 4) sorted by relevance
87 packed_var0 = packed_distributed_variable.PackedVarAndDevice(92 packed_var1 = packed_distributed_variable.PackedVarAndDevice(99 var0 = packed_distributed_variable.PackedVarAndDevice(packed_var, device0)101 var1 = packed_distributed_variable.PackedVarAndDevice(packed_var, device1)
88 return PackedVarAndDevice(self, device)247 class PackedVarAndDevice(object): class353 PackedVarAndDevice, _tensor_conversion_packed_var_and_device)
54 if isinstance(var, packed.PackedVarAndDevice):117 if isinstance(var, packed.PackedVarAndDevice):
916 self.assertIsInstance(val, packed.PackedVarAndDevice)922 self.assertIsInstance(val, packed.PackedVarAndDevice)