Searched defs:table_descriptor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_embedding_load_retrieve_ops.cc | 65 const auto& table_descriptor = config.table_descriptor(table_id); in ComputeExpectedTableShardShapes() local
145 const auto& table_descriptor = config_.table_descriptor(table_id); in GetStateVariables() local
295 const auto& table_descriptor = config_.table_descriptor(table_id); in GetStateVariables() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_configuration_utils.cc | 29 for (const auto& table_descriptor : tpu_embedding_config.table_descriptor()) { in ComputeTotalTagCountForDynamicLearningRates() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 2051 def set_optimization_parameters(self, table_descriptor): argument
2065 def set_optimization_parameters(self, table_descriptor): argument
2142 def set_optimization_parameters(self, table_descriptor): argument
2239 def set_optimization_parameters(self, table_descriptor): argument
2316 def set_optimization_parameters(self, table_descriptor): argument
2409 def set_optimization_parameters(self, table_descriptor): argument
2509 def set_optimization_parameters(self, table_descriptor): argument
2605 def set_optimization_parameters(self, table_descriptor): argument
2682 def set_optimization_parameters(self, table_descriptor): argument
2772 def set_optimization_parameters(self, table_descriptor): argument
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_embedding_shape_util.cc | 80 for (auto& table_descriptor : config.table_descriptor()) { in ComputeTableShapes() local
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_configuration.proto | 26 repeated TableDescriptor table_descriptor = 1; field
|