Searched refs:split_params (Results 1 – 2 of 2) sorted by relevance
65 TfLiteSplitParams split_params; in TEST() local
563 split_params = [566 sharded = embedding_ops.embedding_lookup(split_params, ids).eval()597 split_params = [601 split_params, ids, max_norm=1.0).eval()633 split_params = [637 split_params, ids, max_norm=l2_norm,