Searched defs:TensorDesc (Results 1 – 1 of 1) sorted by relevance
18 struct TORCH_API TensorDesc { struct22 TensorDesc(const at::ScalarType& type, const std::vector<bool>& contiguity) in TensorDesc() argument33 TensorDesc( in TensorDesc() argument39 TensorDesc(const at::Tensor& t) in TensorDesc() function42 TensorDesc(const c10::TensorTypePtr& type) in TensorDesc() function71 bool operator==(const TensorDesc& desc) const { argument79 static size_t hash(const TensorDesc& spec) { in hash() argument