Searched defs:TensorIndex (Results 1 – 6 of 6) sorted by relevance
115 TensorIndex(std::nullopt_t) : type_(TensorIndexType::None) {} in TensorIndex() function118 TensorIndex(at::indexing::EllipsisIndexType) in TensorIndex() function120 TensorIndex(const char* str) : TensorIndex(at::indexing::Ellipsis) { in TensorIndex() function129 TensorIndex(SymInt integer) in TensorIndex() function131 TensorIndex(int64_t integer) : TensorIndex(SymInt(integer)) {} in TensorIndex() function132 TensorIndex(int integer) : TensorIndex(SymInt(integer)) {} in TensorIndex() function136 TensorIndex(T boolean) : boolean_(boolean), type_(TensorIndexType::Boolean) {} in TensorIndex() function139 TensorIndex(Slice slice) in TensorIndex() function143 TensorIndex(Tensor tensor) in TensorIndex() function
101 typedef typename internal::traits<OutputBackward>::Index TensorIndex; typedef386 typedef typename internal::traits<Input>::Index TensorIndex; typedef
94 typedef typename internal::traits<OutputBackward>::Index TensorIndex; typedef396 typedef typename internal::traits<Input>::Index TensorIndex; typedef
57 typedef typename internal::traits<Input>::Index TensorIndex; typedef150 typedef typename internal::traits<Input>::Index TensorIndex; in CuboidMaxPooling() typedef
4 typedef int TensorIndex; typedef
19 TEST(TensorIndexingTest, TensorIndex) { in TEST() argument