Searched refs:InputAndShape (Results 1 – 1 of 1) sorted by relevance
284 struct InputAndShape { struct in tensorflow::grappler::__anonbb66c6470111::ArithmeticNodesGroupOptimizerStage285 InputAndShape(const string& input, const TensorShapeProto& shape) in InputAndShape() argument300 std::vector<InputAndShape> inputs;527 std::unordered_map<string, std::vector<InputAndShape>> shape_sig_to_inputs; in RewriteOptimizedNodesGroup()574 std::deque<InputAndShape> add_ops; in RewriteOptimizedNodesGroup()587 const InputAndShape lhs = add_ops.front(); in RewriteOptimizedNodesGroup()589 const InputAndShape rhs = add_ops.front(); in RewriteOptimizedNodesGroup()594 InputAndShape add = AddAggregatedInputs(*group.root_node, name, lhs, rhs); in RewriteOptimizedNodesGroup()598 InputAndShape optimized_root_node = add_ops.front(); in RewriteOptimizedNodesGroup()603 InputAndShape AddInputsOfSymbolicallyEqualShape( in AddInputsOfSymbolicallyEqualShape()[all …]