Searched refs:original_dtype (Results 1 – 3 of 3) sorted by relevance
362 DataType original_dtype; in RestoreTensorsV2() local365 tensor_name, &original_dtype, &restored_full_shape)); in RestoreTensorsV2()366 if (dtypes[i] != original_dtype) { in RestoreTensorsV2()370 DataTypeString(original_dtype)); in RestoreTensorsV2()
534 original_dtype = self.sparse_types[self.sparse_keys.index(key)]535 if original_dtype != dtype:537 (original_dtype, dtype, key))
4574 def _convert(self, original, original_dtype, output_dtype, expected): argument4575 x_np = np.array(original, dtype=original_dtype.as_numpy_dtype())