Searched defs:paramsDict (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/onnx/ |
| H A D | init.cpp | 80 std::map<std::string, IValue>& paramsDict) { in initONNXBindings()
105 std::map<std::string, IValue>& paramsDict) { in initONNXBindings()
158 std::map<std::string, IValue>& paramsDict) { in initONNXBindings()
167 std::map<std::string, IValue>& paramsDict) { in initONNXBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | unpack_quantized_weights.cpp | 259 std::map<std::string, IValue>& paramsDict, in unpackQuantizedWeightsHelper()
593 std::map<std::string, IValue>& paramsDict) { in UnpackQuantizedWeights()
716 std::map<std::string, IValue>& paramsDict, in insertPermutesHelper()
748 std::map<std::string, IValue>& paramsDict) { in insertPermutes()
|
| H A D | eval_peephole.cpp | 143 void EvalPeepholeONNX(Block* b, ParamMap& paramsDict) { in EvalPeepholeONNX()
149 void EvalPeepholeONNX(std::shared_ptr<Graph>& g, ParamMap& paramsDict) { in EvalPeepholeONNX()
|
| H A D | helper.cpp | 23 const ParamMap& paramsDict) { in buildValueToParamsMap()
56 ParamMap& paramsDict) { in buildParamsMapFromValueToParamsMap()
|
| H A D | constant_fold.cpp | 622 void ConstantFoldONNX(Block* b, ParamMap& paramsDict, int opset_version) { in ConstantFoldONNX()
703 ParamMap& paramsDict, in ConstantFoldONNX()
|
| H A D | eliminate_unused_items.cpp | 10 void EliminateUnusedItemsONNX(Block* b, ParamMap& paramsDict) { in EliminateUnusedItemsONNX()
|
| H A D | deduplicate_initializers.cpp | 85 std::map<std::string, IValue>& paramsDict, in DeduplicateInitializers()
|