Searched defs:TensorArg (Results 1 – 5 of 5) sorted by relevance
22 struct TORCH_API TensorArg { struct27 TensorArg(const Tensor& tensor, const char* name, int pos) in TensorArg() argument
102 …(CheckedFrom c, ArrayRef<TensorArg> tensors, void(*fn)(CheckedFrom, const TensorArg&, const Tensor… in checkAllSame()
103 struct TensorArg { struct105 TensorArg(uint32_t i = 0) : id(i) {} in id() argument122 TensorArg& lookup(const at::Tensor& tensor, bool create = false) { argument
81 def is_unaligned_buffer(arg: TensorArg):
69 class TensorArg: class