Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dconcatenation_test.cc34 std::vector<std::vector<int>> all_input_shapes; in BaseConcatenationOpModel() local
36 all_input_shapes.push_back(input_template.shape); in BaseConcatenationOpModel()
45 BuildInterpreter(all_input_shapes); in BaseConcatenationOpModel()
67 std::vector<std::vector<int>> all_input_shapes; in QuantizedConcatenationOpModel() local
70 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()
79 BuildInterpreter(all_input_shapes); in QuantizedConcatenationOpModel()
Dpack_test.cc31 std::vector<std::vector<int>> all_input_shapes; in PackOpModel() local
33 all_input_shapes.push_back(input_template.shape); in PackOpModel()
40 BuildInterpreter(all_input_shapes); in PackOpModel()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc909 std::vector<std::vector<int>> all_input_shapes; in BaseConcatenationOpModel() local
911 all_input_shapes.push_back(input_template.shape); in BaseConcatenationOpModel()
920 BuildInterpreter(all_input_shapes); in BaseConcatenationOpModel()
965 std::vector<std::vector<int>> all_input_shapes; in QuantizedConcatenationOpModel() local
968 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()
977 BuildInterpreter(all_input_shapes); in QuantizedConcatenationOpModel()
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py1222 all_input_shapes = set(
1224 if len(all_input_shapes) == 1: