Home
last modified time | relevance | path

Searched refs:ValueToParamPairMap (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/onnx/
Dhelper.h20 using ValueToParamPairMap = std::map<Value*, std::pair<std::string, IValue>>; variable
25 const ValueToParamPairMap& valsToParamsMap,
27 TORCH_API ValueToParamPairMap
29 TORCH_API void eraseUnusedValuesFromMap(ValueToParamPairMap& valsToParamsMap);
32 const ValueToParamPairMap& valsToParamsMap,
Dhelper.cpp21 ValueToParamPairMap buildValueToParamsMap( in buildValueToParamsMap()
24 ValueToParamPairMap valsToParamsMap; in buildValueToParamsMap()
43 void eraseUnusedValuesFromMap(ValueToParamPairMap& valsToParamsMap) { in eraseUnusedValuesFromMap()
55 const ValueToParamPairMap& valsToParamsMap, in buildParamsMapFromValueToParamsMap()
Deval_peephole.cpp17 const ValueToParamPairMap& valsToParamsMap) { in getValues()
42 static void fuseConvBatchNorm(Block* b, ValueToParamPairMap& valsToParamsMap) { in fuseConvBatchNorm()
Dconstant_fold.cpp545 bool isConstant(Value* val, const ValueToParamPairMap& valsToParamsMap) { in isConstant()
556 bool hasParamInput(Node* n, const ValueToParamPairMap& valsToParamsMap) { in hasParamInput()
567 const ValueToParamPairMap& valsToParamsMap) { in getValues()
592 const ValueToParamPairMap& valsToParamsMap) { in areNodeInputsConstant()
Ddeduplicate_initializers.cpp15 ValueToParamPairMap& valsToParamsMap, in DeduplicateInitializers()