Searched refs:GetTensorShapeFromXlaArgument (Results 1 – 1 of 1) sorted by relevance
70 StatusOr<TensorShape> GetTensorShapeFromXlaArgument(const XlaArgument& arg) { in GetTensorShapeFromXlaArgument() function539 GetTensorShapeFromXlaArgument(xla_arg)); in RewriteWithArgs()596 GetTensorShapeFromXlaArgument(arg)); in CompileGraphSetup()