Home
last modified time | relevance | path

Searched refs:useInferredTypeMap (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/onnx/
Dconstant_map.cpp22 ConstantValueMap::getInstance().useInferredTypeMap[tensorName] = true; in SetRank()
58 ConstantValueMap::getInstance().useInferredTypeMap[tensorName] = true; in SetShape()
189 ConstantValueMap::getInstance().useInferredTypeMap[tensorName] = value; in SetUseInferredType()
193 return ConstantValueMap::getInstance().useInferredTypeMap.find(tensorName) != in HasUseInferredType()
194 ConstantValueMap::getInstance().useInferredTypeMap.end(); in HasUseInferredType()
202 return ConstantValueMap::getInstance().useInferredTypeMap[tensorName]; in GetUseInferredType()
264 UpdateStrKey<decltype(useInferredTypeMap)>( in UpdateValueName()
265 ConstantValueMap::getInstance().useInferredTypeMap, old_name, new_name); in UpdateValueName()
277 ConstantValueMap::getInstance().useInferredTypeMap.clear(); in ClearMaps()
326 for (const auto& x : ConstantValueMap::getInstance().useInferredTypeMap) { in PrintMaps()
Dconstant_map.h97 std::unordered_map<std::string, bool> useInferredTypeMap; in C10_DIAGNOSTIC_POP() local