Searched refs:input_shapes (Results 1 – 3 of 3) sorted by relevance
92 std::vector<std::vector<uint32_t>> input_shapes(input_shapes0); in LSTMOpModel() local94 input_shapes.push_back({n_batch, n_output}); in LSTMOpModel()95 input_shapes.push_back({n_batch, n_cell}); in LSTMOpModel()96 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()
175 std::vector<std::vector<uint32_t>> input_shapes{ in SVDFOpModel() local183 auto it = input_shapes.begin(); in SVDFOpModel()