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