Home
last modified time | relevance | path

Searched defs:node_kind (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp201 auto node_kind = n->kind(); in IsValidONNXControlflowNode() local
212 auto node_kind = n->kind(); in IsValidONNXNode() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_quant_dequant.cpp159 NodeKind node_kind, in insertQParam()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp475 bool sr_schema_check_kind(torch::jit::Node* node, c10::Symbol node_kind) { in sr_schema_check_kind()