Searched refs:embedding_input_attrs (Results 1 – 1 of 1) sorted by relevance
1042 EmbeddingResourceAttrs embedding_input_attrs; in GetTPUEmbeddingInputNodes() local1046 embedding_input_attrs.table_id = table_id; in GetTPUEmbeddingInputNodes()1053 embedding_input_attrs.slot_id = slot_id; in GetTPUEmbeddingInputNodes()1060 non_sparse_inputs[arg_id - 1]->UpdateAttrs(embedding_input_attrs, s); in GetTPUEmbeddingInputNodes()