Support for Split operation
We support splitting of a tensor along an axis into n pieces, where n has to be a constant. Eg: Split(Tensor of shape(5,30), splits=3, axis=1) returns 3 tensors of shape(5,10) each. Currently we do not suport splitting into tensors of specified shape (num_or_size_splits) though that functionality will be added later. We also do not support splitting into n pieces where n is a runtime value because we do not support run-time code generation yet. This also adds support in the frontend for an op to return multiple values.
Showing
- Athos/SeeDot/AST/AST.py 14 additions, 1 deletionAthos/SeeDot/AST/AST.py
- Athos/SeeDot/AST/ASTVisitor.py 5 additions, 0 deletionsAthos/SeeDot/AST/ASTVisitor.py
- Athos/SeeDot/AST/MtdAST.py 4 additions, 0 deletionsAthos/SeeDot/AST/MtdAST.py
- Athos/SeeDot/AST/PrintAST.py 6 additions, 0 deletionsAthos/SeeDot/AST/PrintAST.py
- Athos/SeeDot/IR/IRBuilderCSF.py 31 additions, 0 deletionsAthos/SeeDot/IR/IRBuilderCSF.py
- Athos/SeeDot/Optimizations/LivenessOpti.py 5 additions, 0 deletionsAthos/SeeDot/Optimizations/LivenessOpti.py
- Athos/SeeDot/Type.py 20 additions, 0 deletionsAthos/SeeDot/Type.py
- Athos/TFCompiler/ProcessTFGraph.py 25 additions, 25 deletionsAthos/TFCompiler/ProcessTFGraph.py
- Athos/TFCompiler/TFNodesAST.py 98 additions, 70 deletionsAthos/TFCompiler/TFNodesAST.py
Loading
Please register or sign in to comment