Searched refs:split_params (Results 1 – 3 of 3) sorted by relevance
66 TfLiteSplitParams split_params; in TEST() local
564 split_params = [567 sharded = embedding_ops.embedding_lookup(split_params, ids)597 split_params = [601 split_params, ids, max_norm=1.0)633 split_params = [637 split_params, ids, max_norm=l2_norm, transform_fn=transform)
1827 const TfLiteSplitVParams* split_params; in IsSupported() local1828 RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &split_params)); in IsSupported()1829 if (split_params->num_splits == 1) { in IsSupported()