Searched defs:PartitionDesc (Results 1 – 1 of 1) sorted by relevance
17 struct TORCH_API PartitionDesc { struct18 PartitionDesc() : nSubTensors_{1}, dim_{0} {} in PartitionDesc() function20 PartitionDesc(const TensorDesc& _desc, size_t _nSubTensors, size_t _dim) in PartitionDesc() function23 std::vector<bool> cont = _desc.contiguity;34 bool isNoop() const { in isNoop()37 size_t nSubTensors() const { in nSubTensors()40 size_t dim() const { in dim()43 std::shared_ptr<TensorDesc> subTensorDesc() { in subTensorDesc()46 const std::shared_ptr<TensorDesc> subTensorDesc() const { in subTensorDesc()53 size_t dim_; // dimension along which the chunk/concat occurs[all …]