Searched refs:ShapeSignature (Results 1 – 1 of 1) sorted by relevance
416 string ShapeSignature(const TensorShapeProto& shape) const { in ShapeSignature() function in tensorflow::grappler::__anon327bfa1e0111::ArithmeticNodesGroupOptimizerStage554 shape_sig_to_inputs[ShapeSignature(input.shape)].push_back(input); in RewriteOptimizedNodesGroup()604 const auto& inputs = shape_sig_to_inputs[ShapeSignature(shapes[i])]; in RewriteOptimizedNodesGroup()984 sigs.insert(ShapeSignature(ias.shape)); in CountUniqueShapes()