pytorch/pytorch/third_party/nvfuser/csrc/python_frontend/ Source Tree

../
fusion_cache.cpp
fusion_cache.h FusionCache FusionCacheEntry
fusion_definition.cpp
fusion_definition.h FusionDefinition Scalar State StateType Tensor
fusion_interface.cpp
fusion_interface.h FusionInterface
fusion_record.h BatchNormOpRecord BroadcastInDimOpRecord BroadcastOpRecord CastOpRecord ConstantRecord EndRecord FullOpRecord NormOpRecord OpRecord OutputRecord PermuteOpRecord RecordFunctor RecordType ReductionOpRecord ScalarRecord SqueezeOpRecord StartRecord TensorRecord VarianceMeanOpRecord VarianceOpRecord ViewOpRecord equal_to hash
python_bindings.cpp
python_bindings.h
python_bindings_extension.cpp
[+] test/

Browse the source code of pytorch/pytorch/third_party/nvfuser/csrc/python_frontend/ online