Searched refs:InputAndShape (Results 1 – 1 of 1) sorted by relevance
309 struct InputAndShape { struct in tensorflow::grappler::__anon327bfa1e0111::ArithmeticNodesGroupOptimizerStage310 InputAndShape(const string& input, const TensorShapeProto& shape) in InputAndShape() function325 std::vector<InputAndShape> inputs;552 std::unordered_map<string, std::vector<InputAndShape>> shape_sig_to_inputs; in RewriteOptimizedNodesGroup()599 std::deque<InputAndShape> add_ops; in RewriteOptimizedNodesGroup()612 const InputAndShape lhs = add_ops.front(); in RewriteOptimizedNodesGroup()614 const InputAndShape rhs = add_ops.front(); in RewriteOptimizedNodesGroup()619 InputAndShape add = AddAggregatedInputs(*group.root_node, name, lhs, rhs); in RewriteOptimizedNodesGroup()623 InputAndShape optimized_root_node = add_ops.front(); in RewriteOptimizedNodesGroup()628 InputAndShape AddInputsOfSymbolicallyEqualShape( in AddInputsOfSymbolicallyEqualShape()[all …]