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()
199 std::vector<std::vector<uint32_t>> output_shapes{{batches_, memory_size_ * units_ * rank_}, in SVDFOpModel() local203 auto it2 = output_shapes.begin(); in SVDFOpModel()224 it2 = output_shapes.begin(); in SVDFOpModel()
117 std::vector<std::vector<uint32_t>> output_shapes{ in LSTMOpModel() local125 auto it2 = output_shapes.begin(); in LSTMOpModel()150 it2 = output_shapes.begin(); in LSTMOpModel()