Searched refs:CheckStateShape (Results 1 – 1 of 1) sorted by relevance
124 Status CheckStateShape(Algorithm alg, const TensorShape& shape) { in CheckStateShape() function193 TF_RETURN_IF_ERROR(CheckStateShape(alg, var_shape)); in CompileImpl()446 OP_REQUIRES_OK(ctx, CheckStateShape(alg, var_shape)); in Compile()