Searched defs:new_dtype (Results 1 – 3 of 3) sorted by relevance
38 def get_dtype_change_hook(new_dtype): argument
753 int new_dtype = desired_np_dtype >= 0 ? desired_np_dtype : array_dtype; in PySeqToTFE_TensorHandle() local
735 …nsize(SymIntArrayRef old_strides, int64_t size_ratio, ScalarType old_dtype, ScalarType new_dtype) { in compute_strides_for_view_dtype_downsize()753 …psize(SymIntArrayRef old_strides, int64_t size_ratio, ScalarType old_dtype, ScalarType new_dtype) { in compute_strides_for_view_dtype_upsize()