Searched defs:TensorIndex (Results 1 – 2 of 2) sorted by relevance
233 explicit TensorIndex(const py::none &) : type_(TensorIndexType::None) {} in TensorIndex() function235 explicit TensorIndex(const py::ellipsis &) : type_(TensorIndexType::Ellipsis) {} in TensorIndex() function237 explicit TensorIndex(int64_t integer) : integer_(integer), type_(TensorIndexType::Integer) {} in TensorIndex() function238 explicit TensorIndex(int integer) : TensorIndex(static_cast<int64_t>(integer)) {} in TensorIndex() function239 explicit TensorIndex(const py::int_ &integer) : TensorIndex(integer.cast<int64_t>()) {} in TensorIndex() function241 explicit TensorIndex(bool boolean) : boolean_(boolean), type_(TensorIndexType::Boolean) {} in TensorIndex() function242 explicit TensorIndex(const py::bool_ &boolean) : TensorIndex(py::cast<bool>(boolean)) {} in TensorIndex() function244 explicit TensorIndex(const Slice &slice) : slice_(slice), type_(TensorIndexType::Slice) {} in TensorIndex() function245 explicit TensorIndex(const py::slice &py_slice) in TensorIndex() function248 …explicit TensorIndex(TensorPtr tensor) : tensor_(std::move(tensor)), type_(TensorIndexType::Tensor… in TensorIndex() function[all …]
44 explicit TensorIndex(const std::string &name) : MetaFuncGraph(name) {} in TensorIndex() function