Searched refs:input_shapes (Results 1 – 3 of 3) sorted by relevance
93 std::vector<std::vector<uint32_t>> input_shapes(input_shapes0); in LSTMOpModel() local95 input_shapes.push_back({n_batch, n_output}); in LSTMOpModel()96 input_shapes.push_back({n_batch, n_cell}); in LSTMOpModel()97 auto it = input_shapes.begin(); in LSTMOpModel()
100 std::vector<std::vector<uint32_t>> input_shapes(input_shapes0); in LayerNormLSTMOpModel() local102 auto it = input_shapes.begin(); in LayerNormLSTMOpModel()106 CHECK(it != input_shapes.end()); \ in LayerNormLSTMOpModel()
173 std::vector<std::vector<uint32_t>> input_shapes{ in SVDFOpModel() local181 auto it = input_shapes.begin(); in SVDFOpModel()