Searched defs:input_shape_name (Results 1 – 2 of 2) sorted by relevance
58 def check_input_shape(input_shape_name, input_shape, enable_none=False): argument
2738 auto input_shape_name = node_name + "input_shape"; in ExportPrimNMSWithMask() local2986 auto input_shape_name = GetNodeInputName(node->input(kThreeNum), node_map_ptr, graph_proto); in ExportPrimScatterNd() local