Searched refs:table_ids (Results 1 – 17 of 17) sorted by relevance
380 table_ids, argument443 table_ids=table_ids,460 table_ids, argument523 table_ids=table_ids,
55 name: "table_ids"59 corresponding input. The ith input is looked up using table_ids[i]. The size60 of the table_ids list must be equal to that of sample_indices,67 to the ith feature. table_ids[i] indicates which embedding table to look up ith
57 name: "table_ids"61 corresponding input. The ith input is looked up using table_ids[i]. The size62 of the table_ids list must be equal to that of sample_indices,69 to the ith feature. table_ids[i] indicates which embedding table to look up ith
44 name: "table_ids"131 name: "table_ids"218 name: "table_ids"313 name: "table_ids"
83 name: "table_ids"178 name: "table_ids"
44 name: "table_ids"131 name: "table_ids"218 name: "table_ids"
83 name: "table_ids"
20 …", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0], table_ids = [1, 1, 0]} : (…48 …", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0], table_ids = [1, 1, 0]} : (…61 …", "sum"], device_ordinal = -1 : i64, max_sequence_lengths = [0, 0, 0], table_ids = [1, 1, 0]} : (…
199 … "tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf.string>) -> ()200 … "tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf.string>) -> ()
486 …eddingRaggedTensorBatch"(%arg0) {_xla_outside_compilation = "cluster1", table_ids = [1, 2]} : (ten…
983 table_ids = []1000 table_ids.append(table_to_id[feature.table])1024 table_ids=table_ids,1033 table_ids=table_ids,
1349 …', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids\', \'device_ordi…1357 …', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids\', \'device_ordi…
48796 … []tf.Output, aggregation_weights []tf.Output, mode_override tf.Output, table_ids []int64, optiona…48800 attrs := map[string]interface{}{"table_ids": table_ids}52102 … []tf.Output, aggregation_weights []tf.Output, mode_override tf.Output, table_ids []int64, optiona…52106 attrs := map[string]interface{}{"table_ids": table_ids}
3745 to the ith feature. table_ids[i] indicates which embedding table to look up ith3772 I64ArrayAttr:$table_ids,3835 to the ith feature. table_ids[i] indicates which embedding table to look up ith3860 I64ArrayAttr:$table_ids,
14092 name: "table_ids"14278 name: "table_ids"
22380 name: "table_ids"22467 name: "table_ids"