Home
last modified time | relevance | path

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

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