Home
last modified time | relevance | path

Searched refs:table_descriptor (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py1449 table_descriptor = config_proto.table_descriptor.add()
1450 table_descriptor.name = table
1455 table_descriptor.vocabulary_size = max(table_config.vocabulary_size,
1457 table_descriptor.dimension = table_config.dimension
1459 table_descriptor.num_features = self._table_to_num_features_dict[table]
1464 parameters = table_descriptor.optimization_parameters
1501 optimizer_handler.set_optimization_parameters(table_descriptor)
2010 def set_optimization_parameters(self, table_descriptor): argument
2024 def set_optimization_parameters(self, table_descriptor): argument
2025 table_descriptor.optimization_parameters.adagrad.SetInParent()
[all …]
Dtpu_embedding_v2.py478 table_descriptor = config_proto.table_descriptor.add()
479 table_descriptor.name = table.name
483 table_descriptor.vocabulary_size = max(table.vocabulary_size,
485 table_descriptor.dimension = table.dim
487 table_descriptor.num_features = num_features[table]
489 parameters = table_descriptor.optimization_parameters
Dtpu_embedding_v2_utils_test.py99 td = config.table_descriptor.add()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.cc28 config.table_descriptor()) { in ComputeOutputTensorShapes()
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dtpu_embedding_configuration.proto23 repeated TableDescriptor table_descriptor = 1; field
53 // of hosts, each of the first "table_descriptor.vocabulary_size % num_hosts"