Home
last modified time | relevance | path

Searched refs:all_input_shapes (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dconcatenation_test.cc41 std::vector<std::vector<int>> all_input_shapes; in BaseConcatenationOpModel() local
44 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() local
497 all_input_shapes.push_back(input_template[i].shape); in PersistentConcatenationOpModel()
506 BuildInterpreter(all_input_shapes, /*num_threads=*/-1, in PersistentConcatenationOpModel()
H A Dpack_test.cc36 std::vector<std::vector<int>> all_input_shapes; in PackOpModel() local
38 all_input_shapes.push_back(input_template.shape); in PackOpModel()
45 BuildInterpreter(all_input_shapes); in PackOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dpack_test.cc26 std::vector<std::vector<int>> all_input_shapes; in PackOpModel() local
28 all_input_shapes.push_back(input_template.shape); in PackOpModel()
35 BuildInterpreter(all_input_shapes); in PackOpModel()
H A Dconcat_test.cc51 std::vector<std::vector<int>> all_input_shapes; in QuantizedConcatenationOpModel() local
53 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()
62 BuildInterpreter(all_input_shapes); in QuantizedConcatenationOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc1812 std::vector<std::vector<int>> all_input_shapes; in BaseConcatenationOpModel() local
1814 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() local
1871 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()
1880 BuildInterpreterWithNNAPI(all_input_shapes); in QuantizedConcatenationOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dbase_layer_v1.py1599 all_input_shapes = set(
1601 if len(all_input_shapes) == 1:
H A Dbase_layer.py2116 all_input_shapes = set(
2118 if len(all_input_shapes) == 1: