Searched defs:valsToParamsMap (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | constant_fold.cpp | 545 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()
630 auto valsToParamsMap = buildValueToParamsMap(b, paramsDict); in ConstantFoldONNX() local
|
| H A D | eval_peephole.cpp | 17 const ValueToParamPairMap& valsToParamsMap) { in getValues()
42 static void fuseConvBatchNorm(Block* b, ValueToParamPairMap& valsToParamsMap) { in fuseConvBatchNorm()
144 auto valsToParamsMap = buildValueToParamsMap(b, paramsDict); in EvalPeepholeONNX() local
|
| H A D | helper.cpp | 24 ValueToParamPairMap valsToParamsMap; in buildValueToParamsMap() local
43 void eraseUnusedValuesFromMap(ValueToParamPairMap& valsToParamsMap) { in eraseUnusedValuesFromMap()
55 const ValueToParamPairMap& valsToParamsMap, in buildParamsMapFromValueToParamsMap()
|
| H A D | deduplicate_initializers.cpp | 15 ValueToParamPairMap& valsToParamsMap, in DeduplicateInitializers()
87 auto valsToParamsMap = buildValueToParamsMap(g->block(), paramsDict); in DeduplicateInitializers() local
|
| H A D | eliminate_unused_items.cpp | 11 auto valsToParamsMap = buildValueToParamsMap(b, paramsDict); in EliminateUnusedItemsONNX() local
|
| H A D | unpack_quantized_weights.cpp | 538 auto valsToParamsMap = buildValueToParamsMap(b, paramsDict); in unpackQuantizedWeightsHelper() local
|
| H A D | shape_type_inference.cpp | 1882 auto valsToParamsMap = buildValueToParamsMap(b, params_dict); in ONNXShapeTypeInference() local
|