Searched refs:AddRewritesForShape (Results 1 – 1 of 1) sorted by relevance
127 Status AddRewritesForShape(int i, const xla::Shape& shape, in AddRewritesForShape() function180 AddRewritesForShape(i, xla::Shape(ps.parameters(i)), &rewrites)); in GenArgMethods()229 TF_RETURN_IF_ERROR(AddRewritesForShape( in GenResultMethods()262 AddRewritesForShape(i, xla::Shape(ps.parameters(i)), &rewrites)); in GenVariableMethods()