| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 23 Status ComputeOutputTensorShapes(const TPUEmbeddingConfiguration& config, in ComputeOutputTensorShapes()
27 for (const TPUEmbeddingConfiguration::TableDescriptor& table : in ComputeOutputTensorShapes()
|
| D | tpu_embedding_output_layout_utils.h | 28 const TPUEmbeddingConfiguration& config,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt | 15 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
17 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
|
| D | api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 33 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
35 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
60 (offset of TableDescriptor in the TPUEmbeddingConfiguration) to lookup the
|
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 31 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
33 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
58 (offset of TableDescriptor in the TPUEmbeddingConfiguration) to lookup the
|
| D | api_def_ConfigureTPUEmbedding.pbtxt | 7 Serialized tensorflow.tpu.TPUEmbeddingConfiguration that
|
| D | api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt | 31 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
33 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
|
| D | api_def_ConfigureDistributedTPU.pbtxt | 20 Serialized tensorflow.tpu.TPUEmbeddingConfiguration that
|
| D | api_def_RecvTPUEmbeddingActivations.pbtxt | 21 Serialized TPUEmbeddingConfiguration proto.
|
| D | api_def_SendTPUEmbeddingGradients.pbtxt | 29 Serialized TPUEmbeddingConfiguration proto.
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_ops.cc | 69 tpu::TPUEmbeddingConfiguration config; in __anon355f22aa0102()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_utils_test.py | 97 config = tpu_embedding_configuration_pb2.TPUEmbeddingConfiguration()
|
| D | tpu_embedding_v2.py | 448 ) -> tpu_embedding_configuration_pb2.TPUEmbeddingConfiguration:
457 config_proto = tpu_embedding_configuration_pb2.TPUEmbeddingConfiguration()
505 tpu_embedding_configuration_pb2.TPUEmbeddingConfiguration.TRAINING)
513 tpu_embedding_configuration_pb2.TPUEmbeddingConfiguration.DIV_DEFAULT)
|
| D | tpu_embedding.py | 47 TRAINING = elc.TPUEmbeddingConfiguration.TRAINING
48 INFERENCE = elc.TPUEmbeddingConfiguration.INFERENCE
1447 config_proto = elc.TPUEmbeddingConfiguration()
1508 elc.TPUEmbeddingConfiguration.DIV_DEFAULT
1510 elc.TPUEmbeddingConfiguration.MOD)
|
| D | tpu_embedding_v2_utils.py | 740 config: tpu_embedding_configuration_pb2.TPUEmbeddingConfiguration) -> None: argument
|
| D | tpu.py | 114 embedding_config: Optional[embedding_pb2.TPUEmbeddingConfiguration] = None, argument
154 embedding_config: embedding_pb2.TPUEmbeddingConfiguration, argument
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_output_layout.proto | 6 // proto contained in the TPUEmbeddingConfiguration.
|
| D | tpu_embedding_configuration.proto | 8 message TPUEmbeddingConfiguration { message
|
| D | optimization_parameters.proto | 13 // Dynamic learning rate specification in the TPUEmbeddingConfiguration. The
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 941 config: Serialized TPUEmbeddingConfiguration proto.
964 TPUEmbeddingConfiguration proto given to tpu.initialize_system.
970 TPUEmbeddingConfiguration proto.
976 config: Serialized TPUEmbeddingConfiguration proto.
|
| D | tf_generated_ops.td | 3728 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
3730 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
3766 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
3768 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
3812 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
3814 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
3854 TPUEmbeddingConfiguration. Supported values are {'unspecified', 'inference',
3856 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
|