Home
last modified time | relevance | path

Searched refs:float_dtype (Results 1 – 14 of 14) sorted by relevance

/external/pytorch/torch/_numpy/
D_dtypes.py270 arg = _dtypes_impl.default_dtypes().float_dtype
396 float_dtype = torch.float64
398 float_dtype = torch.float32
400 float_dtype = fp_dtype
406 }[float_dtype]
414 float_dtype=float_dtype, complex_dtype=complex_dtype, int_dtype=int_dtype
Drandom.py86 dtype = _dtypes_impl.default_dtypes().float_dtype
105 dtype = _dtypes_impl.default_dtypes().float_dtype
112 dtype = _dtypes_impl.default_dtypes().float_dtype
121 dtype = _dtypes_impl.default_dtypes().float_dtype
D_dtypes_impl.py30 float_dtype=getattr(torch, config.numpy_default_float),
34 assert isinstance(_default_dtypes.float_dtype, torch.dtype)
47 return default_dtypes().float_dtype
D_reductions_impl.py62 return _dtypes_impl.default_dtypes().float_dtype
394 dtype = _dtypes_impl.default_dtypes().float_dtype
431 q = q.to(_dtypes_impl.default_dtypes().float_dtype)
Dlinalg.py21 a = a.to(_dtypes_impl.default_dtypes().float_dtype)
28 dtyp = _dtypes_impl.default_dtypes().float_dtype
D_funcs_impl.py307 dtype = _dtypes_impl.default_dtypes().float_dtype
368 _dtypes_impl.default_dtypes().float_dtype
398 dtype = _dtypes_impl.default_dtypes().float_dtype
459 dtype = _dtypes_impl.default_dtypes().float_dtype
484 dtype = _dtypes_impl.default_dtypes().float_dtype
1056 dtype = _dtypes_impl.default_dtypes().float_dtype
1815 dtype = _dtypes_impl.default_dtypes().float_dtype
1820 dtype = _dtypes_impl.default_dtypes().float_dtype
1825 dtype = _dtypes_impl.default_dtypes().float_dtype
1830 dtype = _dtypes_impl.default_dtypes().float_dtype
[all …]
D_util.py42 x = x.to(_dtypes_impl.default_dtypes().float_dtype)
Dfft.py21 else _dtypes_impl.default_dtypes().float_dtype
/external/executorch/extension/pybindings/test/
Dmake_test.py323 float_dtype = 6
340 [t.dtype() for t in input_tensors], [float_dtype, float_dtype]
349 tester.assertEqual(output_tensor.dtype(), float_dtype)
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dreaders.py63 def _is_valid_float(str_val, float_dtype): argument
65 return float_dtype.as_numpy_dtype(str_val) < np.inf
/external/pytorch/test/
Dtest_unary_ufuncs.py707 float_dtype = (
710 np_float_out = np_fn(a).astype(torch_to_numpy_dtype_dict[float_dtype])
711 float_out = torch.empty_like(t, dtype=float_dtype)
716 float_out = torch.empty(1, device=device, dtype=float_dtype)
/external/pytorch/torch/csrc/inductor/aoti_torch/
Dshim_common.cpp1108 at::ScalarType float_dtype = in aoti_torch_print_tensor_handle() local
1110 std::cout << "Mean value: " << mean_value(float_dtype) << std::endl; in aoti_torch_print_tensor_handle()
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py732 float_dtype = np_dtypes.default_float_type()
733 a = a.astype(float_dtype)
734 factor = math_ops.cast(factor, float_dtype)
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dlookup_ops_test.py1643 for float_dtype in [dtypes.float32, dtypes.float64]:
1645 values = constant_op.constant([0.0, 1.1, 2.2, 3.3], float_dtype)
1646 default_value = constant_op.constant(-1.5, float_dtype)
1649 float_dtype,