Searched refs:output_shapes (Results 1 – 3 of 3) sorted by relevance
125 std::vector<std::vector<uint32_t>> output_shapes{ in LayerNormLSTMOpModel() local133 auto it2 = output_shapes.begin(); in LayerNormLSTMOpModel()136 CHECK(it2 != output_shapes.end()); \ in LayerNormLSTMOpModel()159 it2 = output_shapes.begin(); in LayerNormLSTMOpModel()
201 std::vector<std::vector<uint32_t>> output_shapes{{batches_, memory_size_ * units_ * rank_}, in SVDFOpModel() local205 auto it2 = output_shapes.begin(); in SVDFOpModel()226 it2 = output_shapes.begin(); in SVDFOpModel()
116 std::vector<std::vector<uint32_t>> output_shapes{ in LSTMOpModel() local124 auto it2 = output_shapes.begin(); in LSTMOpModel()149 it2 = output_shapes.begin(); in LSTMOpModel()