Searched defs:default_dtype (Results 1 – 13 of 13) sorted by relevance
| /external/pytorch/test/cpp/api/ |
| D | tensor.cpp | 365 c10::ScalarType default_dtype) { in test_TorchTensorCtorScalarFloatingType_expected_dtype()
451 c10::ScalarType default_dtype) { in test_TorchTensorCtorSingleDimFloatingType_expected_dtype()
588 c10::ScalarType default_dtype) { in test_TorchTensorCtorMultiDimFloatingType_expected_dtype()
735 c10::ScalarType default_dtype) { in test_TorchTensorCtorMultiDim_CUDA_expected_dtype()
762 c10::ScalarType default_dtype) { in test_TorchTensorCtorZeroSizedDim_expected_dtype()
824 c10::ScalarType default_dtype) { in test_TorchTensorCtorWithoutSpecifyingDtype_expected_dtype()
852 c10::ScalarType default_dtype) { in test_TorchTensorCtorWithNonDtypeOptions_expected_dtype()
899 void test_Arange_expected_dtype(c10::ScalarType default_dtype) { in test_Arange_expected_dtype()
|
| /external/pytorch/c10/core/ |
| D | DefaultDtype.cpp | 5 static auto default_dtype = caffe2::TypeMeta::Make<float>(); variable
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | stateful_random_ops.cc | 35 #define REGISTER_STATEFUL_OP(name, default_dtype) \ argument
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_spec.py | 277 def to_tensor_spec(input_spec, default_dtype=None): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter.cc | 160 DataType default_dtype = DT_INVALID) { in ConvertToTensorInPlace()
376 DataType default_dtype = i < input.default_dtypes.size() in ConvertInputsWithTypeListAttr() local
|
| D | python_api_info.h | 96 DataType default_dtype; // DT_INVALID if no default. member
|
| D | python_api_info.cc | 252 DataType default_dtype = attr_def.has_default_value() in InitializeAttribute() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execute.py | 225 def args_to_matching_eager(l, ctx, allowed_dtypes, default_dtype=None): argument
|
| /external/pytorch/torch/_inductor/codegen/cuda/ |
| D | cuda_kernel.py | 199 def cutlass_dtype(self, node: IRNode, default_dtype="void") -> Optional[str]: argument
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | shape_analysis.cpp | 1034 auto default_dtype = in PropagateTensorShapeOnNode() local
1400 [](Node* node, int dim, at::ScalarType default_dtype) -> type_vec_t { in PropagateTensorShapeOnNode()
1917 auto default_dtype = in PropagateCompleteShapeOnNode() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReduceOps.cpp | 1482 auto default_dtype = at::typeMetaToScalarType(c10::get_default_dtype()); in logsumexp_out() local
1496 auto default_dtype = at::typeMetaToScalarType(c10::get_default_dtype()); in logsumexp() local
|
| /external/pytorch/test/ |
| D | test_transformers.py | 174 def calculate_nt_tolerances(nt_ref_hp, nt_ref_lp, default_dtype, fudge_factor=1): argument
|
| D | test_tensor_creation_ops.py | 2168 def test_inference(default_dtype): argument
|