Searched full:nestedtensorcpu (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 288 NestedTensorCPU, NestedTensorCUDA: native_dropout_nested
294 CPU, NestedTensorCPU, NestedTensorCUDA: native_dropout_backward
342 NestedTensorCPU, NestedTensorCUDA: NestedTensor_abs
352 NestedTensorCPU, NestedTensorCUDA: NestedTensor_abs_
429 NestedTensorCPU, NestedTensorCUDA: NestedTensor_sgn
438 NestedTensorCPU, NestedTensorCUDA: NestedTensor_sgn_
556 NestedTensorCPU, NestedTensorCUDA: NestedTensor_add_Tensor
567 NestedTensorCPU, NestedTensorCUDA: NestedTensor_add__Tensor
1248 NestedTensorCPU, NestedTensorCUDA: NestedTensor_logical_not
1256 NestedTensorCPU, NestedTensorCUDA: NestedTensor_logical_not_
[all …]
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKey.cpp | 329 {"NestedTensorCPU", c10::DispatchKey::NestedTensorCPU}, in parseDispatchKey()
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 53 NestedTensorCPU, NestedTensorCUDA: NestedTensor_add_Tensor
77 NestedTensorCPU, NestedTensorCUDA: NestedTensor_mul_Tensor
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 195 NestedTensorCPU = auto() variable in DispatchKey
290 DispatchKey.NestedTensorCPU,
|