Home
last modified time | relevance | path

Searched defs:tensorlist (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_foreach.py1391 def inplace_func(*tensorlist):
1405 def outplace_func(*tensorlist):
H A Dtest_jit.py331 def _sum_of_list(tensorlist): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUFallback.cpp79 static bool validate_tensor_list(const c10::List<at::Tensor>& tensorlist) { in validate_tensor_list()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DVariableTypeUtils.h126 at::ITensorListRef tensorlist, in throw_error_for_complex_autograd()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_arg_parser.h443 inline std::vector<at::Tensor> PythonArgs::tensorlist(int i) { in tensorlist() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.cc276 bool CanInferTensorListElementType(Value tensorlist, in CanInferTensorListElementType()