Searched refs:found_shapes (Results 1 – 1 of 1) sorted by relevance
165 std::unordered_map<string, TensorShape> found_shapes; in CalculateFlops() local167 GetOutputShapes(inputs, wanted_shapes, session, &found_shapes)); in CalculateFlops()179 const TensorShape& filter_shape = found_shapes[node.input(1)]; in CalculateFlops()180 const TensorShape& output_shape = found_shapes[node.name()]; in CalculateFlops()189 const TensorShape& a_shape = found_shapes[node.input(0)]; in CalculateFlops()190 const TensorShape& output_shape = found_shapes[node.name()]; in CalculateFlops()200 const TensorShape& filter_shape = found_shapes[node.input(1)]; in CalculateFlops()201 const TensorShape& output_shape = found_shapes[node.name()]; in CalculateFlops()