• Scott Cyphers's avatar
    TensorIterator (#3038) · 4a25881e
    Scott Cyphers authored
    * TensorIterator
    
    * ssize_t is not on windows
    
    * RNN building test
    
    * simplify
    
    * Simplify output
    
    * typo
    
    * typos
    
    * remove arg
    
    * Sequence version
    
    * style
    
    * Serialization for all but TensorIterator
    
    * Add ops for igpu
    
    * style
    
    * typo, ngpu
    
    * missing headers, output vector
    
    * Fix const json issues
    
    * TensorIterator serialization
    
    * Serialization for TensorIterator
    Switch Outout<T> to use shared_ptr do nodes don't vanish
    Switch Result to new node style
    Add serialization/deserialization to test
    
    * Switch Output to use a shared_ptr to prevent nodes from disappearing early.
    
    * Eliminate wrapped enum
    Switch allreduce to new op form
    
    * Convert to new op form
    
    * Disambiguate concat
    
    * Add autobroadcast for SequencePush
    Add validation for SequencePush
    
    * compute shapes for SequenceRepeat
    
    * Add explicit conversion from PartialShape to dimension vector
    validate and infer types for SliceInput
    
    * validate and infer types for SequenceOutput
    
    * Add sequence attributes
    
    * Move test to serializer so it doesn't fail when there is no serializer?
    
    * const arg
    
    * Beginning of TensorIterator validation
    
    * Validation up to parameters
    
    * Fix shape in test
    
    * Remove mis-typed AxisSet
    
    * Simplify, add doc
    
    * Review comments
    
    * Tweaks
    
    * free/bound
    
    * Try fused op
    
    * Discussion
    
    * more
    
    * comments
    
    * Start of LSTMCell test
    
    * Add LSTMCell example
    
    * Reorg
    
    * Reorg
    
    * Fused ops don't need handlers
    
    * Serialization
    
    * Use `as_type` and `is_type` for up-conversions of descriptions
    Allocate output space for each output
    
    * Clean up type checking
    
    * Fix ser/deser issues
    
    * Refactor, cleanup type info to make it safer to use for non-ops
    
    * Implement validate_and_infer_types and modify unit tests.
    
    * For ops in the loop body: revalidate and infer types.
    
    Nested loop is not supported.
    
    * Put body ops in a set and call revalidate and infer types on the set.
    
    * Set slice[axis] to part_size.
    
    Call set_partial_shape to set shape for body parameters.
    
    Add more unit tests.
    
    * Give tensor iterator body a lambda
    
    * Update validate_and_infer_types and unit tests.
    
    * Serialization of body
    
    * Change static function to TensorIterator function.
    
    * review comments
    4a25881e
Name
Last commit
Last update
.ci Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
.clang-format Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.travis.yml Loading commit data...
ABOUT.md Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIB.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...
ecosystem-overview.md Loading commit data...