Home
last modified time | relevance | path

Searched refs:MergeInferredTypeAndSetMap (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.h33 void MergeInferredTypeAndSetMap(
H A Dshape_type_inference.cpp75 void MergeInferredTypeAndSetMap( in MergeInferredTypeAndSetMap() function
186 MergeInferredTypeAndSetMap(v, v->type(), torch_tensor_type); in UpdateTorchValueByOnnxValueInfo()
192 MergeInferredTypeAndSetMap(v, v->type(), torch_list_type); in UpdateTorchValueByOnnxValueInfo()
2284 MergeInferredTypeAndSetMap( in ONNXUpdateTypeFromTensor()
2355 MergeInferredTypeAndSetMap( in ONNXAssignOutputShape()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Donnx.cpp361 MergeInferredTypeAndSetMap( in NodeToONNX()