Searched defs:TensorOptions (Results 1 – 2 of 2) sorted by relevance
139 struct C10_API TensorOptions { struct140 TensorOptions() in TensorOptions() argument151 /* implicit */ TensorOptions(Layout layout) : TensorOptions() { in TensorOptions() function160 /* implicit */ TensorOptions(T&& device) : TensorOptions() { in TensorOptions() function175 /* implicit */ TensorOptions(Args&&... args) in TensorOptions() function179 /* implicit */ TensorOptions(caffe2::TypeMeta dtype) : TensorOptions() { in TensorOptions() argument184 /* implicit */ TensorOptions(ScalarType dtype) : TensorOptions() { in TensorOptions() function189 /* implicit */ TensorOptions(MemoryFormat memory_format) : TensorOptions() { in TensorOptions() function197 TensorOptions r = *this; in device() argument206 C10_NODISCARD TensorOptions device(Args&&... args) const noexcept { in device() argument[all …]
120 operator TensorOptions() const { in TensorOptions() function