| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 287 mode_override=None, argument
306 if mode_override is None:
307 mode_override = "unspecified"
311 mode_override=mode_override,
325 mode_override=None, argument
360 if mode_override is None:
361 mode_override = "unspecified"
368 mode_override=mode_override,
385 mode_override=None, argument
437 if mode_override is None:
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 19 name: "mode_override"
67 name: "mode_override"
154 name: "mode_override"
249 name: "mode_override"
|
| D | EnqueueTPUEmbeddingSparseBatch.pbtxt | 19 name: "mode_override"
63 name: "mode_override"
|
| D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 19 name: "mode_override"
114 name: "mode_override"
|
| D | EnqueueTPUEmbeddingIntegerBatch.pbtxt | 9 name: "mode_override"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 940 mode_override: Text
1021 mode_override=mode_override,
1030 mode_override=mode_override,
1272 mode_override = array_ops.where_v2(training,
1280 mode_override=mode_override)
1293 mode_override = "train" if training else "inference"
1312 device_ordinal=device_ordinal, mode_override=mode_override)
1320 mode_override = "train" if training else "inference"
1329 mode_override=mode_override)
|
| D | tpu_embedding.py | 1628 mode_override=None, argument
1653 mode_override=mode_override,
1747 mode_override=None, argument
1757 mode_override=mode_override,
1763 mode_override=mode_override,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 19 name: "mode_override"
67 name: "mode_override"
154 name: "mode_override"
|
| D | EnqueueTPUEmbeddingSparseBatch.pbtxt | 19 name: "mode_override"
63 name: "mode_override"
|
| D | EnqueueTPUEmbeddingIntegerBatch.pbtxt | 9 name: "mode_override"
|
| D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 19 name: "mode_override"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingIntegerBatch.pbtxt | 12 name: "mode_override"
17 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
|
| D | api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt | 28 name: "mode_override"
33 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
|
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 28 name: "mode_override"
33 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
|
| D | api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 30 name: "mode_override"
35 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1345 …argspec: "args=[\'batch\', \'mode_override\', \'device_ordinal\', \'name\'], varargs=None, keyword…
1349 …[\'sample_splits\', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids…
1353 …\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'device_or…
1357 …\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1345 …argspec: "args=[\'batch\', \'mode_override\', \'device_ordinal\', \'name\'], varargs=None, keyword…
1349 …[\'sample_splits\', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids…
1353 …\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'device_or…
1357 …\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 40494 …tput, embedding_indices []tf.Output, aggregation_weights []tf.Output, mode_override tf.Output, opt…
40505 …mple_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_override,
47460 func EnqueueTPUEmbeddingIntegerBatch(scope *Scope, batch []tf.Output, mode_override tf.Output, opti…
47471 tf.OutputList(batch), mode_override,
48796 …tput, embedding_indices []tf.Output, aggregation_weights []tf.Output, mode_override tf.Output, tab…
48807 …mple_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_override,
52102 …tput, embedding_indices []tf.Output, aggregation_weights []tf.Output, mode_override tf.Output, tab…
52113 …ample_splits), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_override,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 912 TF_StrTensor:$mode_override,
|
| D | tf_generated_ops.td | 3730 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
3768 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
3814 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
3856 in TPUEmbeddingConfiguration is used, otherwise mode_override is used.}]>:$mode_override,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13992 name: "mode_override"
14028 name: "mode_override"
14131 name: "mode_override"
14214 name: "mode_override"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 22192 name: "mode_override"
22228 name: "mode_override"
22272 name: "mode_override"
22355 name: "mode_override"
22403 name: "mode_override"
|