Searched refs:embedding (Results 1 – 25 of 256) sorted by relevance
1234567891011
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_configuration.proto | 9 // Description of the various embedding tables.
15 // The embedding dimension (i.e., the width of the embedding table).
19 // Details of the learning algorithm used to update the embedding
25 // Mode. Should the embedding layer program be run for inference (just forward
35 // Number of samples in each batch of embedding layer activations sent to
45 // Sharding strategy of the embedding tables among the hosts.
71 // backward pass on the sparse core is executed only after the embedding
80 // core allowing it to process step N+1 while the embedding gradients for step
83 // is complete. The drawback is that embedding activations for step N+1 do not
84 // observe the embedding gradient updates from step N. This could affect model
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RecvTPUEmbeddingActivations.pbtxt | 7 A TensorList of embedding activations containing one Tensor per
8 embedding table in the model.
15 embedding tables in the model.
24 summary: "An op that receives embedding activations on the TPU."
26 The TPU system performs the embedding lookups and aggregations specified by
|
| D | api_def_SendTPUEmbeddingGradients.pbtxt | 7 A TensorList of gradients with which to update embedding tables.
10 with respect to the embedding activations. The embedding tables are updated
11 from these gradients via the optimizer specified in the TPU embedding
32 summary: "Performs gradient updates of embedding tables."
|
| D | api_def_RetrieveTPUEmbeddingStochasticGradientDescentParameters.pbtxt | 10 summary: "Retrieve SGD embedding parameters."
12 An op that retrieves optimization parameters from embedding to host
14 the correct embedding table configuration. For example, this op is
|
| D | api_def_LoadTPUEmbeddingStochasticGradientDescentParameters.pbtxt | 10 summary: "Load SGD embedding parameters."
12 An op that loads optimization parameters into HBM for embedding. Must be
14 embedding table configuration. For example, this op is used to install
|
| D | api_def_RetrieveTPUEmbeddingAdagradParameters.pbtxt | 16 summary: "Retrieve Adagrad embedding parameters."
18 An op that retrieves optimization parameters from embedding to host
20 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingMomentumParameters.pbtxt | 16 summary: "Retrieve Momentum embedding parameters."
18 An op that retrieves optimization parameters from embedding to host
20 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt | 17 summary: "Retrieve frequency estimator embedding parameters."
19 An op that retrieves optimization parameters from embedding to host
21 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt | 16 summary: "Retrieve SGD embedding parameters with debug support."
18 An op that retrieves optimization parameters from embedding to host
20 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingProximalAdagradParameters.pbtxt | 16 summary: "Retrieve proximal Adagrad embedding parameters."
18 An op that retrieves optimization parameters from embedding to host
20 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingAdadeltaParameters.pbtxt | 22 summary: "Retrieve Adadelta embedding parameters."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_LoadTPUEmbeddingProximalAdagradParameters.pbtxt | 16 summary: "Load proximal Adagrad embedding parameters."
18 An op that loads optimization parameters into HBM for embedding. Must be
20 embedding table configuration. For example, this op is used to install
|
| D | api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt | 16 summary: "Load frequency estimator embedding parameters."
18 An op that loads optimization parameters into HBM for embedding. Must be
20 embedding table configuration. For example, this op is used to install
|
| D | api_def_LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.pbtxt | 16 summary: "Load SGD embedding parameters."
18 An op that loads optimization parameters into HBM for embedding. Must be
20 embedding table configuration. For example, this op is used to install
|
| D | api_def_RetrieveTPUEmbeddingRMSPropParameters.pbtxt | 22 summary: "Retrieve RMSProp embedding parameters."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingADAMParameters.pbtxt | 22 summary: "Retrieve ADAM embedding parameters."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingFTRLParameters.pbtxt | 22 summary: "Retrieve FTRL embedding parameters."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_LoadTPUEmbeddingMomentumParameters.pbtxt | 16 summary: "Load Momentum embedding parameters."
18 An op that loads optimization parameters into HBM for embedding. Must be
20 embedding table configuration. For example, this op is used to install
|
| D | api_def_LoadTPUEmbeddingAdagradParameters.pbtxt | 16 summary: "Load Adagrad embedding parameters."
18 An op that loads optimization parameters into HBM for embedding. Must be
20 embedding table configuration. For example, this op is used to install
|
| D | api_def_RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.pbtxt | 22 summary: "Retrieve proximal Adagrad embedding parameters with debug support."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.pbtxt | 22 summary: "Retrieve Adagrad embedding parameters with debug support."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt | 26 "Retrieve frequency estimator embedding parameters with debug support."
28 An op that retrieves optimization parameters from embedding to host
30 the correct embedding table configuration. For example, this op is
|
| D | api_def_RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt | 22 summary: "Retrieve Momentum embedding parameters with debug support."
24 An op that retrieves optimization parameters from embedding to host
26 the correct embedding table configuration. For example, this op is
|
| D | api_def_LoadTPUEmbeddingMomentumParametersGradAccumDebug.pbtxt | 22 summary: "Load Momentum embedding parameters with debug support."
24 An op that loads optimization parameters into HBM for embedding. Must be
26 embedding table configuration. For example, this op is used to install
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 261 embedding = embedding_ops.embedding_lookup(p, ids)
263 tf_result = embedding.eval(feed_dict=feed_dict)
266 self.assertShapeEqual(np_result, embedding)
274 embedding = embedding_ops.embedding_lookup(
277 self.assertAllEqual(embedding, [[1.0]])
285 embedding = embedding_ops.embedding_lookup(
291 self.assertAllEqual(embedding, 2 * self.evaluate(normalized))
304 embedding = embedding_ops.embedding_lookup(p_variable, ids)
310 tf_result = embedding.eval(feed_dict=feed_dict)
314 self.assertShapeEqual(np_result, embedding)
[all …]
|
1234567891011