Home
last modified time | relevance | path

Searched defs:TensorNames (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorNames.h52 struct TORCH_API TensorNames { struct
70 explicit TensorNames(TensorNameVec&& names) : names_(std::move(names)){}; in TensorNames() function
72 TensorNameVec names_;
H A DTensorNames.cpp48 TensorNames::TensorNames(ArrayRef<Dimname> names) { in TensorNames() function in at::namedinference::TensorNames
55 TensorNames::TensorNames(ArrayRef<Dimname> names, int64_t start, int64_t end) { in TensorNames() function in at::namedinference::TensorNames