| /external/pytorch/aten/src/ATen/native/ |
| D | MetaTensor.cpp | 16 std::optional<ScalarType> dtype_opt, in empty_meta_symint()
34 std::optional<ScalarType> dtype_opt, in empty_strided_meta_symint()
|
| D | Constraints.cpp | 81 std::optional<ScalarType> dtype_opt, in _make_dep_token_cpu()
|
| D | TensorFactories.cpp | 247 Tensor empty_cpu(IntArrayRef size, std::optional<ScalarType> dtype_opt, std::optional<Layout> layou… in empty_cpu()
280 …muted_symint(SymIntArrayRef size, IntArrayRef physical_layout, std::optional<ScalarType> dtype_opt, in empty_permuted_symint()
319 Tensor empty_strided_cpu(IntArrayRef size, IntArrayRef stride, std::optional<ScalarType> dtype_opt, in empty_strided_cpu()
503 std::optional<ScalarType> dtype_opt, in new_empty_symint()
1180 int64_t row, int64_t col, int64_t offset, std::optional<ScalarType> dtype_opt, in tril_indices_cpu()
1230 int64_t row, int64_t col, int64_t offset, std::optional<ScalarType> dtype_opt, in triu_indices_cpu()
1441 std::optional<ScalarType> dtype_opt, in bartlett_window()
1480 std::optional<ScalarType> dtype_opt, in blackman_window()
1551 std::optional<ScalarType> dtype_opt, in hamming_window()
1628 std::optional<ScalarType> dtype_opt, in kaiser_window()
|
| /external/pytorch/aten/src/ATen/ |
| D | EmptyTensor.cpp | 265 std::optional<ScalarType> dtype_opt, in empty_cpu()
300 std::optional<ScalarType> dtype_opt, in empty_strided_cpu()
356 std::optional<ScalarType> dtype_opt, in empty_meta()
376 std::optional<ScalarType> dtype_opt, in empty_symint_meta()
410 std::optional<ScalarType> dtype_opt, in empty_strided_meta()
445 std::optional<ScalarType> dtype_opt, in empty_strided_symint_meta()
|
| D | ScalarOps.cpp | 26 Tensor scalar_tensor_static(const Scalar& s, std::optional<ScalarType> dtype_opt, std::optional<Dev… in scalar_tensor_static()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TensorFactories.cu | 54 Tensor empty_cuda(IntArrayRef size, std::optional<ScalarType> dtype_opt, std::optional<Layout> layo… in empty_cuda()
80 Tensor empty_strided_cuda(IntArrayRef size, IntArrayRef stride, std::optional<ScalarType> dtype_opt… in empty_strided_cuda()
277 int64_t row, int64_t col, int64_t offset, std::optional<ScalarType> dtype_opt, in tril_indices_cuda()
353 int64_t row, int64_t col, int64_t offset, std::optional<ScalarType> dtype_opt, in triu_indices_cuda()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | EmptyTensor.cpp | 25 std::optional<ScalarType> dtype_opt, in empty_cuda()
66 std::optional<ScalarType> dtype_opt, in empty_strided_cuda()
|
| /external/pytorch/aten/src/ATen/native/mps/ |
| D | TensorFactory.cpp | 77 std::optional<ScalarType> dtype_opt, in empty_mps()
89 std::optional<ScalarType> dtype_opt, in empty_strided_mps()
|
| /external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/ |
| D | OpenRegMem.cpp | 63 std::optional<c10::ScalarType> dtype_opt, in empty_openreg()
82 std::optional<c10::ScalarType> dtype_opt, in empty_strided_openreg()
|
| /external/pytorch/aten/src/ATen/mps/ |
| D | EmptyTensor.cpp | 23 std::optional<ScalarType> dtype_opt, in empty_mps()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | codegen.h | 85 std::optional<c10::ScalarType> dtype_opt, in empty_strided()
|
| D | cuda_codegen.cpp | 1253 std::optional<c10::ScalarType> dtype_opt, in empty_strided()
|
| D | llvm_codegen.cpp | 449 std::optional<c10::ScalarType> dtype_opt, in empty_strided()
|
| /external/pytorch/test/inductor/extension_backends/cpp/ |
| D | extension_device.cpp | 137 …_strided(c10::IntArrayRef size, c10::IntArrayRef stride, std::optional<at::ScalarType> dtype_opt, … in custom_empty_strided()
|
| /external/pytorch/c10/core/ |
| D | TensorOptions.h | 310 std::optional<caffe2::TypeMeta> dtype_opt() const noexcept { in dtype_opt() function
|
| /external/pytorch/test/cpp_extensions/ |
| D | open_registration_extension.cpp | 386 std::optional<at::ScalarType> dtype_opt, in custom_empty_strided()
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | ops.cpp | 1426 const auto dtype_opt = p_node->Input(1).toOptional<at::ScalarType>(); in call() local
|