Lines Matching full:symint
7 #include <c10/core/SymInt.h>
27 constexpr int64_t INDEX_MIN = c10::SymInt::min_representable_int();
30 enum class TensorIndexType { None, Ellipsis, SymInt, Boolean, Slice, Tensor }; enumerator
42 std::optional<c10::SymInt> start_index = std::nullopt,
43 std::optional<c10::SymInt> stop_index = std::nullopt,
44 std::optional<c10::SymInt> step_index = std::nullopt) {
46 step_ = c10::SymInt(1);
56 start_ = c10::SymInt(step_ < 0 ? INDEX_MAX : 0);
62 stop_ = c10::SymInt(step_ < 0 ? INDEX_MIN : INDEX_MAX);
68 inline c10::SymInt start() const { in start()
72 inline c10::SymInt stop() const { in stop()
76 inline c10::SymInt step() const { in step()
81 c10::SymInt start_;
82 c10::SymInt stop_;
83 c10::SymInt step_;
129 TensorIndex(SymInt integer) in TensorIndex()
130 : integer_(std::move(integer)), type_(TensorIndexType::SymInt) {} in TensorIndex()
131 TensorIndex(int64_t integer) : TensorIndex(SymInt(integer)) {} in TensorIndex()
132 TensorIndex(int integer) : TensorIndex(SymInt(integer)) {} in TensorIndex()
155 return type_ == TensorIndexType::SymInt; in is_integer()
158 inline SymInt integer() const { in integer()
187 SymInt integer_ = 0;
205 c10::SymInt start, in applySlice()
206 c10::SymInt stop, in applySlice()
207 c10::SymInt step, in applySlice()
221 SymInt length = (self_device == at::kCPU || self_device == at::kCUDA) in applySlice()
237 SymInt index, in applySelect()
395 // Unbacked SymInt has different behavior, but this is sound because in slicePrefix1sSize()