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