| /external/ml_dtypes/ml_dtypes/tests/ |
| D | intn_test.py | 57 def testModuleName(self, scalar_type): argument
61 def testPickleable(self, scalar_type): argument
73 def testRoundTripToPythonScalar(self, scalar_type, python_scalar): argument
82 def testRoundTripNumpyTypes(self, scalar_type): argument
95 def testStr(self, scalar_type): argument
100 def testRepr(self, scalar_type): argument
105 def testItem(self, scalar_type): argument
110 def testHash(self, scalar_type): argument
125 def testComparison(self, scalar_type, op): argument
139 def testUnop(self, scalar_type, op): argument
[all …]
|
| /external/pytorch/torch/csrc/utils/ |
| D | tensor_new.cpp | 55 at::ScalarType scalar_type, in build_options()
71 at::ScalarType scalar_type, in new_with_sizes()
81 at::ScalarType scalar_type, in new_with_storage()
88 std::vector<int64_t> compute_sizes(PyObject* seq, ScalarType scalar_type) { in compute_sizes()
292 at::ScalarType scalar_type, in internal_new_from_data()
501 at::ScalarType scalar_type, in new_from_data_copy()
516 at::ScalarType scalar_type, in legacy_new_from_sequence()
603 at::ScalarType scalar_type, in legacy_sparse_tensor_generic_ctor_new()
707 at::ScalarType scalar_type, in legacy_tensor_generic_ctor_new()
826 at::ScalarType scalar_type, in legacy_tensor_ctor()
[all …]
|
| D | tensor_types.cpp | 103 auto scalar_type = torch::tensors::get_default_scalar_type(); in options_from_string() local
172 for (auto& scalar_type : scalar_types) { in all_declared_types() local
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | scalar_type_analysis.cpp | 89 const c10::ScalarType& scalar_type) { in CreateProfiledTensorTypeWithScalarType()
160 [&typesFromTensors, &typesFromScalars](at::ScalarType scalar_type) { in InferExpectedScalarType()
215 auto scalar_type = tensor.scalar_type(); in InferExpectedScalarType() local
222 } else if (auto scalar_type = get_scalar_type(input)) { in InferExpectedScalarType() local
284 const c10::ScalarType& scalar_type) { in LowPrecisionCastForStandardOps()
299 const c10::ScalarType& scalar_type) { in UpdateScalarTypeForInputs()
356 const c10::ScalarType& scalar_type) { in UpdateScalarTypeForOutput()
365 const c10::ScalarType& scalar_type) { in RecoverScalarTypeForOutput()
|
| /external/pytorch/torch/csrc/ |
| D | Dtype.cpp | 15 PyObject* THPDtype_New(at::ScalarType scalar_type, const std::string& name) { in THPDtype_New()
80 auto scalar_type = self->scalar_type; in THPDtype_to_real() local
91 auto scalar_type = self->scalar_type; in THPDtype_to_complex() local
|
| D | Dtype.h | 10 PyObject_HEAD at::ScalarType scalar_type; member
|
| /external/pytorch/aten/src/ATen/core/ |
| D | tensor_type.cpp | 277 std::optional<at::ScalarType> scalar_type, in create()
307 std::optional<at::ScalarType> scalar_type, in create()
319 std::optional<at::ScalarType> scalar_type, in create()
358 auto scalar_type = merge_primitive(scalarType(), other.scalarType()); in merge() local
434 std::optional<at::ScalarType> scalar_type, in TensorType()
449 at::ScalarType scalar_type, in createContiguous()
|
| D | QuantizerBase.h | 59 ScalarType scalar_type() const { in scalar_type() function
|
| /external/executorch/kernels/portable/cpu/test/ |
| D | scalar_utils_test.cpp | 31 #define CALL_TEST_ONE(cpp_type, scalar_type) \ in testAll() argument
68 #define INSTANTIATE_TYPE_TEST(cpp_type, scalar_type) \ in TEST() argument
|
| /external/executorch/runtime/executor/ |
| D | method_meta.cpp | 57 exec_aten::ScalarType scalar_type) { in calculate_nbytes()
71 exec_aten::ScalarType scalar_type, in TensorInfo()
87 exec_aten::ScalarType TensorInfo::scalar_type() const { in scalar_type() function in executorch::runtime::TensorInfo
|
| D | tensor_parser_portable.cpp | 40 ScalarType scalar_type = static_cast<ScalarType>(s_tensor->scalar_type()); in parseTensor() local
|
| /external/pytorch/aten/src/ATen/ |
| D | EmptyTensor.cpp | 167 ScalarType scalar_type, in _empty_generic()
201 ScalarType scalar_type, in empty_generic()
210 ScalarType scalar_type, in empty_generic_symint()
221 ScalarType scalar_type) { in _empty_strided_generic()
243 ScalarType scalar_type) { in empty_strided_generic()
252 ScalarType scalar_type) { in empty_strided_symint_generic()
384 auto scalar_type = dtype_or_default(dtype_opt); in empty_symint_meta() local
|
| D | Utils.h | 33 ScalarType scalar_type) { in checked_dense_tensor_list_unwrap()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDADataType.h | 55 inline cudaDataType ScalarTypeToCudaDataType(const c10::ScalarType& scalar_type) { in ScalarTypeToCudaDataType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 112 PrimitiveType scalar_type() const { in scalar_type() function
125 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anonfae897050111::GemvConfig
139 explicit GemvConfig(std::string name, PrimitiveType scalar_type, in GemvConfig()
226 explicit Config(PrimitiveType scalar_type, int64_t tile_rows, in Config()
454 explicit Config(PrimitiveType scalar_type, int64_t tile_rows, in Config()
665 explicit Config(PrimitiveType scalar_type, Dimensions dims, in Config()
684 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anonfae897050111::TiledSmallGemmEmitter::Config
762 PrimitiveType scalar_type() const { return config().scalar_type(); } in scalar_type() function in xla::cpu::__anonfae897050111::TiledSmallGemmEmitter
1002 void EmitRowMajorGemv(PrimitiveType scalar_type, int64_t tile_rows, in EmitRowMajorGemv()
1029 void EmitColumnMajorGemv(PrimitiveType scalar_type, int64_t tile_rows, in EmitColumnMajorGemv()
[all …]
|
| /external/pytorch/c10/core/ |
| D | ScalarTypeToTypeMeta.h | 16 inline caffe2::TypeMeta scalarTypeToTypeMeta(ScalarType scalar_type) { in scalarTypeToTypeMeta()
|
| /external/pytorch/torch/csrc/jit/backends/coreml/objc/ |
| D | PTMCoreMLTensorSpec.h | 16 static inline c10::ScalarType scalar_type(const std::string& type_string) { in scalar_type() function
|
| /external/executorch/runtime/core/exec_aten/util/test/ |
| D | scalar_type_util_test.cpp | 182 #define CALL_TEST_ONE(cpp_type, scalar_type) \ in testAll() argument
217 #define INSTANTIATE_TYPE_TEST(cpp_type, scalar_type) \ in TEST() argument
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | graph_opt.cpp | 110 auto scalar_type = in doesCatPromoteTypes() local
307 std::optional<at::ScalarType> scalar_type, in fixupTypeInfoForValue()
343 std::optional<at::ScalarType> scalar_type; in inferScalarType() local
397 std::optional<at::ScalarType> scalar_type = inferScalarType(n); in fixupMissingShapeInfo() local
|
| /external/pytorch/aten/src/ATen/quantized/ |
| D | Quantizer.cpp | 43 ScalarType scalar_type) { in make_per_tensor_affine_quantizer()
52 ScalarType scalar_type) { in make_per_channel_affine_quantizer()
146 auto scalar_type = typeMetaToScalarType(dtype); in new_qtensor() local
450 QuantizerPtr make_unknown_quantizer(ScalarType scalar_type) { in make_unknown_quantizer()
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | shape.h | 26 c10::ScalarType scalar_type() const { in scalar_type() function
|
| /external/pytorch/torch/csrc/autograd/ |
| D | variable_info.h | 15 at::ScalarType scalar_type = at::kFloat; member
|
| /external/pytorch/torch/csrc/api/include/torch/detail/ |
| D | TensorDataContainer.h | 31 inline c10::ScalarType compute_desired_dtype(c10::ScalarType scalar_type) { in compute_desired_dtype()
240 const c10::ScalarType& scalar_type() const { in scalar_type() function
|
| /external/executorch/exir/verification/ |
| D | bindings.cpp | 58 const std::vector<int64_t>& strides) { in PYBIND11_MODULE()
|
| /external/pytorch/torch/csrc/tensor/ |
| D | python_tensor.cpp | 42 int scalar_type; member
355 ScalarType scalar_type = declared_types[i].second; in initialize_aten_types() local
451 auto scalar_type = ((THPDtype*)obj)->scalar_type; in py_set_default_dtype() local
|