Searched refs:shape_compute_function_name (Results 1 – 1 of 1) sorted by relevance
247 const std::string& shape_compute_function_name, in genShapeComputeFn() argument255 shape_compute_function_name); in genShapeComputeFn()256 if (reused_functions.count(shape_compute_function_name)) { in genShapeComputeFn()258 graph = reused_functions[shape_compute_function_name]; in genShapeComputeFn()261 module.get_function(shape_compute_function_name); in genShapeComputeFn()269 reused_functions[shape_compute_function_name] = graph; in genShapeComputeFn()279 const std::string& shape_compute_function_name, in registerSchema() argument283 schema_string, shape_compute_function_name, reused_functions, module); in registerSchema()314 const std::string& shape_compute_function_name = pair.second; in loadModule() local317 schema_string, shape_compute_function_name, reused_functions, module); in loadModule()[all …]