Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/onnx/
Dconstant_map.h59 static void SetUseInferredType( in C10_DIAGNOSTIC_POP()
Dconstant_map.cpp186 void ConstantValueMap::SetUseInferredType( in SetUseInferredType() function in torch::jit::ConstantValueMap
Dshape_type_inference.cpp81 ConstantValueMap::SetUseInferredType(dest_v->debugName(), inferred); in MergeInferredTypeAndSetMap()
2166 ConstantValueMap::SetUseInferredType(node_output->debugName(), true); in ONNXShapeTypeInference()