Home
last modified time | relevance | path

Searched defs:TensorArg (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/
DTensorUtils.h22 struct TORCH_API TensorArg { struct
27 TensorArg(const Tensor& tensor, const char* name, int pos) in TensorArg() function
DTensorUtils.cpp102 …(CheckedFrom c, ArrayRef<TensorArg> tensors, void(*fn)(CheckedFrom, const TensorArg&, const Tensor… in checkAllSame()
/external/pytorch/torch/_inductor/codegen/
Dtriton_utils.py81 def is_unaligned_buffer(arg: TensorArg):
Dcommon.py69 class TensorArg: class
/external/pytorch/torch/csrc/dynamo/
Dcompiled_autograd.h103 struct TensorArg { struct
105 TensorArg(uint32_t i = 0) : id(i) {} in id() argument
122 TensorArg& lookup(const at::Tensor& tensor, bool create = false) { argument