Searched refs:GetOutputShapes (Results 1 – 3 of 3) sorted by relevance
57 std::vector<std::vector<int>> GetOutputShapes() { in GetOutputShapes() function in tflite::__anon05b8f2d30111::UnpackOpModel77 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()97 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()115 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()133 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()153 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()169 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()188 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()209 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()228 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST()[all …]
83 std::vector<std::vector<int>> GetOutputShapes() { in GetOutputShapes() function in tflite::ops::experimental::__anonb9dfc1660111::CTCBeamSearchDecoderOpModel105 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST()137 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST()169 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST()210 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST()
116 Status GetOutputShapes(const std::vector<InputLayerInfo>& inputs, in GetOutputShapes() function167 GetOutputShapes(inputs, wanted_shapes, session, &found_shapes)); in CalculateFlops()