Searched refs:shape_sig_to_inputs (Results 1 – 1 of 1) sorted by relevance
527 std::unordered_map<string, std::vector<InputAndShape>> shape_sig_to_inputs; in RewriteOptimizedNodesGroup() local529 shape_sig_to_inputs[ShapeSignature(input.shape)].push_back(input); in RewriteOptimizedNodesGroup()532 using SigKV = decltype(shape_sig_to_inputs)::value_type; in RewriteOptimizedNodesGroup()533 VLOG(3) << "Add/AddN group has " << shape_sig_to_inputs.size() in RewriteOptimizedNodesGroup()535 << str_util::Join(shape_sig_to_inputs, ", ", in RewriteOptimizedNodesGroup()542 shapes.reserve(shape_sig_to_inputs.size()); in RewriteOptimizedNodesGroup()543 for (const auto& el : shape_sig_to_inputs) in RewriteOptimizedNodesGroup()579 const auto& inputs = shape_sig_to_inputs[ShapeSignature(shapes[i])]; in RewriteOptimizedNodesGroup()