Searched refs:embedding_pb2 (Results 1 – 1 of 1) sorted by relevance
34 from tensorflow.core.protobuf.tpu import tpu_embedding_configuration_pb2 as embedding_pb2 unknown114 embedding_config: Optional[embedding_pb2.TPUEmbeddingConfiguration] = None,154 embedding_config: embedding_pb2.TPUEmbeddingConfiguration,