Searched refs:type_is_device_array (Results 1 – 3 of 3) sorted by relevance
859 return [_copy_device_array_to_device(x, device) if type_is_device_array(x)1049 def type_is_device_array(x): function1287 return type_is_device_array(x) and type(x.device_buffer) is DeviceConstant1382 if type_is_device_array(x):
312 all(xla.type_is_device_array(x) for x in out_flat) and
3267 if xla.type_is_device_array(operand) and np.all(3575 if xla.type_is_device_array(operand) and dimensions is None:3694 if xla.type_is_device_array(operand):