FM / tensorexpr

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr

Name	Size	Mode	Actions
operators/	-	0755	rm
analysis.h	5888	0644	edit dl rm
block_codegen.h	4211	0644	edit dl rm
bounds_inference.h	2230	0644	edit dl rm
bounds_overlap.h	3329	0644	edit dl rm
codegen.h	6402	0644	edit dl rm
cpp_codegen.h	2278	0644	edit dl rm
cpp_intrinsics.h	719	0644	edit dl rm
cuda_codegen.h	7782	0644	edit dl rm
cuda_random.h	2642	0644	edit dl rm
dim_arg.h	884	0644	edit dl rm
eval.h	9639	0644	edit dl rm
exceptions.h	3253	0644	edit dl rm
expr.h	11588	0644	edit dl rm
external_functions.h	1274	0644	edit dl rm
external_functions_registry.h	2343	0644	edit dl rm
fwd_decls.h	2806	0644	edit dl rm
graph_opt.h	2553	0644	edit dl rm
half_support.h	5038	0644	edit dl rm
hash_provider.h	7930	0644	edit dl rm
intrinsic_symbols.h	420	0644	edit dl rm
ir.h	22622	0644	edit dl rm
ir_cloner.h	2069	0644	edit dl rm
ir_mutator.h	2010	0644	edit dl rm
ir_printer.h	3693	0644	edit dl rm
ir_simplifier.h	15090	0644	edit dl rm
ir_verifier.h	1240	0644	edit dl rm
ir_visitor.h	1825	0644	edit dl rm
kernel.h	9210	0644	edit dl rm
llvm_codegen.h	3180	0644	edit dl rm
llvm_jit.h	1965	0644	edit dl rm
loopnest.h	21599	0644	edit dl rm
mem_dependency_checker.h	13003	0644	edit dl rm
reduction.h	6742	0644	edit dl rm
registerizer.h	12498	0644	edit dl rm
stmt.h	21138	0644	edit dl rm
tensor.h	7640	0644	edit dl rm
tensorexpr_init.h	268	0644	edit dl rm
types.h	3880	0644	edit dl rm
unique_name_manager.h	940	0644	edit dl rm
var_substitutor.h	1753	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/ir_mutator.h (2010B)

#pragma once
#include <c10/core/ScalarType.h>
#include <torch/csrc/WindowsTorchApiMacro.h>
#include <torch/csrc/jit/tensorexpr/fwd_decls.h>
#include <vector>

namespace torch {
namespace jit {
namespace tensorexpr {

class TORCH_API IRMutator {
 public:
  virtual ~IRMutator() = default;
  virtual ExprPtr mutate(AddPtr v);
  virtual ExprPtr mutate(SubPtr v);
  virtual ExprPtr mutate(MulPtr v);
  virtual ExprPtr mutate(DivPtr v);
  virtual ExprPtr mutate(ModPtr v);
  virtual ExprPtr mutate(MaxPtr v);
  virtual ExprPtr mutate(MinPtr v);
  virtual ExprPtr mutate(AndPtr v);
  virtual ExprPtr mutate(OrPtr v);
  virtual ExprPtr mutate(XorPtr v);
  virtual ExprPtr mutate(LshiftPtr v);
  virtual ExprPtr mutate(RshiftPtr v);
  virtual ExprPtr mutate(CompareSelectPtr v);
#define IMM_MUTATE_DECLARE(Type, Name) virtual ExprPtr mutate(Name##ImmPtr v);
  AT_FORALL_SCALAR_TYPES_AND3(Bool, Half, BFloat16, IMM_MUTATE_DECLARE);
#undef IMM_MUTATE_DECLARE
  virtual ExprPtr mutate(CastPtr v);
  virtual ExprPtr mutate(BitCastPtr v);
  virtual ExprPtr mutate(VarPtr v);
  virtual ExprPtr mutate(BufPtr v);
  virtual ExprPtr mutate(RampPtr v);
  virtual ExprPtr mutate(LoadPtr v);
  virtual ExprPtr mutate(BroadcastPtr v);
  virtual ExprPtr mutate(IfThenElsePtr v);
  virtual ExprPtr mutate(IntrinsicsPtr v);

virtual ExprPtr mutate(TermPtr v);
  virtual ExprPtr mutate(PolynomialPtr v);
  virtual ExprPtr mutate(RoundOffPtr v);
  virtual ExprPtr mutate(MaxTermPtr v);
  virtual ExprPtr mutate(MinTermPtr v);

virtual ExprPtr mutate(ReduceOpPtr v);

virtual StmtPtr mutate(ForPtr v);
  virtual StmtPtr mutate(BlockPtr v);
  virtual StmtPtr mutate(StorePtr v);
  virtual StmtPtr mutate(AtomicAddPtr v);
  virtual StmtPtr mutate(SyncThreadsPtr v);
  virtual StmtPtr mutate(ExternalCallPtr v);

virtual StmtPtr mutate(AllocatePtr v);
  virtual StmtPtr mutate(FreePtr v);
  virtual StmtPtr mutate(LetPtr v);
  virtual StmtPtr mutate(CondPtr v);
};

} // namespace tensorexpr
} // namespace jit
} // namespace torch

#pragma once #include #include #include #include namespace torch { namespace jit { namespace tensorexpr { class TORCH_API IRMutator { public: virtual ~IRMutator() = default; virtual ExprPtr mutate(AddPtr v); virtual ExprPtr mutate(SubPtr v); virtual ExprPtr mutate(MulPtr v); virtual ExprPtr mutate(DivPtr v); virtual ExprPtr mutate(ModPtr v); virtual ExprPtr mutate(MaxPtr v); virtual ExprPtr mutate(MinPtr v); virtual ExprPtr mutate(AndPtr v); virtual ExprPtr mutate(OrPtr v); virtual ExprPtr mutate(XorPtr v); virtual ExprPtr mutate(LshiftPtr v); virtual ExprPtr mutate(RshiftPtr v); virtual ExprPtr mutate(CompareSelectPtr v); #define IMM_MUTATE_DECLARE(Type, Name) virtual ExprPtr mutate(Name##ImmPtr v); AT_FORALL_SCALAR_TYPES_AND3(Bool, Half, BFloat16, IMM_MUTATE_DECLARE); #undef IMM_MUTATE_DECLARE virtual ExprPtr mutate(CastPtr v); virtual ExprPtr mutate(BitCastPtr v); virtual ExprPtr mutate(VarPtr v); virtual ExprPtr mutate(BufPtr v); virtual ExprPtr mutate(RampPtr v); virtual ExprPtr mutate(LoadPtr v); virtual ExprPtr mutate(BroadcastPtr v); virtual ExprPtr mutate(IfThenElsePtr v); virtual ExprPtr mutate(IntrinsicsPtr v); virtual ExprPtr mutate(TermPtr v); virtual ExprPtr mutate(PolynomialPtr v); virtual ExprPtr mutate(RoundOffPtr v); virtual ExprPtr mutate(MaxTermPtr v); virtual ExprPtr mutate(MinTermPtr v); virtual ExprPtr mutate(ReduceOpPtr v); virtual StmtPtr mutate(ForPtr v); virtual StmtPtr mutate(BlockPtr v); virtual StmtPtr mutate(StorePtr v); virtual StmtPtr mutate(AtomicAddPtr v); virtual StmtPtr mutate(SyncThreadsPtr v); virtual StmtPtr mutate(ExternalCallPtr v); virtual StmtPtr mutate(AllocatePtr v); virtual StmtPtr mutate(FreePtr v); virtual StmtPtr mutate(LetPtr v); virtual StmtPtr mutate(CondPtr v); }; } // namespace tensorexpr } // namespace jit } // namespace torch