Home
last modified time | relevance | path

Searched defs:vocabulary_size (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_embedding_shape_util.cc38 int64 vocabulary_size, int table_dimension, int shard_id, int num_shards, in ComputeOneTableShape()
96 TpuEmbeddingShapeUtil::ComputeNumEmbeddingIdsPerShard(int64 vocabulary_size, in ComputeNumEmbeddingIdsPerShard()
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dtpu_embedding_configuration.proto14 int64 vocabulary_size = 2; field
/external/tensorflow/tensorflow/python/feature_column/
Dsequence_feature_column.py183 key, vocabulary_file, vocabulary_size=None, num_oov_buckets=0, argument
Dfeature_column_v2.py1245 vocabulary_size=None, argument
1351 vocabulary_size=None, argument
3488 vocabulary_size, argument
Dfeature_column.py1161 vocabulary_size=None, argument
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py60 vocabulary_size, argument
2993 vocabulary_size, argument