| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convert.h | 84 c10::scalarTypeToTypeMeta(convert_dtype(tensor.dtype())), in convert()
95 c10::scalarTypeToTypeMeta(convert_dtype(tensor.dtype())), in convert_quantized()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | jit_utils.h | 73 auto result_size = c10::scalarTypeToTypeMeta(desc.result_type).itemsize(); in can_vectorize_up_to()
77 auto input_size = c10::scalarTypeToTypeMeta(desc.f_inputs_type).itemsize(); in can_vectorize_up_to()
|
| H A D | SpectralOps.cu | 102 const int64_t element_size = scalarTypeToTypeMeta(dtype).itemsize(); in _fft_fill_with_conjugate_symmetry_cuda_()
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | support.h | 176 torch::set_default_dtype(torch::scalarTypeToTypeMeta(default_dtype)); in AutoDefaultDtypeMode()
180 torch::set_default_dtype(torch::scalarTypeToTypeMeta(prev_default_dtype)); in ~AutoDefaultDtypeMode()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | ScalarTypeToTypeMeta.h | 16 inline caffe2::TypeMeta scalarTypeToTypeMeta(ScalarType scalar_type) { in scalarTypeToTypeMeta() function
|
| H A D | TensorOptions.h | 491 dtype_ = scalarTypeToTypeMeta(*dtype); in set_dtype()
584 return TensorOptions().dtype(scalarTypeToTypeMeta(dtype)); in dtype()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | EmptyTensor.cpp | 171 caffe2::TypeMeta dtype = scalarTypeToTypeMeta(scalar_type); in _empty_generic()
224 caffe2::TypeMeta dtype = scalarTypeToTypeMeta(scalar_type); in _empty_strided_generic()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | DeprecatedTypeProperties.h | 54 return scalarTypeToTypeMeta(scalar_type_); in typeMeta()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ComplexHelper.h | 27 c10::TensorImpl::VIEW, std::move(storage), key_set, scalarTypeToTypeMeta(dtype)); in view_tensor()
|
| H A D | UnaryOps.cpp | 888 auto dtype = c10::scalarTypeToTypeMeta(self.scalar_type()); in mvlgamma()
|
| H A D | TensorConversions.cpp | 778 const auto type_meta = c10::scalarTypeToTypeMeta(dtype); in view_dtype()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/mps/ |
| H A D | EmptyTensor.cpp | 49 auto dtype_meta = scalarTypeToTypeMeta(dtype); in empty_mps()
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | maia_extension.cpp | 26 return get_tensor(scalarTypeToTypeMeta(dtype_or_default(dtype)), size); in empty_override()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Dtype.cpp | 42 scalarTypeToTypeMeta(self->scalar_type).itemsize()); in THPDtype_itemsize()
|
| /aosp_15_r20/external/executorch/extension/tensor/ |
| H A D | tensor_impl_ptr.cpp | 112 .dtype(c10::scalarTypeToTypeMeta(type)) in make_tensor_impl_ptr()
|
| H A D | tensor_ptr.cpp | 111 .dtype(c10::scalarTypeToTypeMeta(type)) in make_tensor_ptr()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | tensor_impl.cpp | 85 c10::scalarTypeToTypeMeta(tensor.dtype()),
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | MKLDNNCommon.cpp | 97 caffe2::TypeMeta dtype_ = scalarTypeToTypeMeta(dtype_or_default(dtype)); in new_with_itensor_mkldnn()
|
| /aosp_15_r20/external/pytorch/torch/csrc/tensor/ |
| H A D | python_tensor.cpp | 339 at::set_default_dtype(scalarTypeToTypeMeta(*dtype)); in set_default_tensor_type()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/ |
| H A D | SpectralOps.cpp | 134 const auto element_size = scalarTypeToTypeMeta(dtype).itemsize(); in _fft_fill_with_conjugate_symmetry_cpu_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensor.cpp | 167 scalarTypeToTypeMeta(dtype_or_default(dtype))); in new_sparse()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | profiler_kineto.cpp | 108 dtypes[i] = std::string(scalarTypeToTypeMeta(t.dtype_).name()); in parseArgData()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | external_functions.cpp | 62 auto typeMeta = c10::scalarTypeToTypeMeta(dtype); in from_blob_quantized()
|
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | Module.cpp | 1270 auto meta = scalarTypeToTypeMeta(toScalarType(dtype_arg)); in registerCudaPluggableAllocator()
|