Searched refs:all_input_shapes (Results 1 – 7 of 7) sorted by relevance
41 std::vector<std::vector<int>> all_input_shapes; in BaseConcatenationOpModel() local44 all_input_shapes.push_back(input_template[i].shape); in BaseConcatenationOpModel()53 BuildInterpreter(all_input_shapes); in BaseConcatenationOpModel()493 std::vector<std::vector<int>> all_input_shapes; in PersistentConcatenationOpModel() local497 all_input_shapes.push_back(input_template[i].shape); in PersistentConcatenationOpModel()506 BuildInterpreter(all_input_shapes, /*num_threads=*/-1, in PersistentConcatenationOpModel()
36 std::vector<std::vector<int>> all_input_shapes; in PackOpModel() local38 all_input_shapes.push_back(input_template.shape); in PackOpModel()45 BuildInterpreter(all_input_shapes); in PackOpModel()
26 std::vector<std::vector<int>> all_input_shapes; in PackOpModel() local28 all_input_shapes.push_back(input_template.shape); in PackOpModel()35 BuildInterpreter(all_input_shapes); in PackOpModel()
51 std::vector<std::vector<int>> all_input_shapes; in QuantizedConcatenationOpModel() local53 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()62 BuildInterpreter(all_input_shapes); in QuantizedConcatenationOpModel()
1812 std::vector<std::vector<int>> all_input_shapes; in BaseConcatenationOpModel() local1814 all_input_shapes.push_back(input_template.shape); in BaseConcatenationOpModel()1823 BuildInterpreterWithNNAPI(all_input_shapes); in BaseConcatenationOpModel()1868 std::vector<std::vector<int>> all_input_shapes; in QuantizedConcatenationOpModel() local1871 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()1880 BuildInterpreterWithNNAPI(all_input_shapes); in QuantizedConcatenationOpModel()
1599 all_input_shapes = set(1601 if len(all_input_shapes) == 1:
2116 all_input_shapes = set(2118 if len(all_input_shapes) == 1: