Home
last modified time | relevance | path

Searched refs:EnqueueTPUEmbeddingSparseTensorBatch (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_update_embedding_enqueue_op_inputs.mlir19 …// CHECK: "tf.EnqueueTPUEmbeddingSparseTensorBatch"(%[[ARG_0]], %[[ARG_1]], %[[ARG_2]], %[[ARG_3]]…
20 …"tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %0, %0, %0, %a…
47 …// CHECK: "tf.EnqueueTPUEmbeddingSparseTensorBatch"(%[[ARG_0]], %[[ARG_1]], %[[ARG_2]], %[[ARG_3]]…
48 …"tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %0, %0, %0, %a…
60 …// expected-error @+1 {{'tf.EnqueueTPUEmbeddingSparseTensorBatch' op must have a corresponding 'tf…
61 …"tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %0, %0, %0, %1…
Dreplicate_to_island.mlir199 … "tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf.string>) -> ()
212 // CHECK: "tf.EnqueueTPUEmbeddingSparseTensorBatch"
219 // CHECK: "tf.EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DEnqueueTPUEmbeddingSparseTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingSparseTensorBatch"
50 name: "EnqueueTPUEmbeddingSparseTensorBatch"
137 name: "EnqueueTPUEmbeddingSparseTensorBatch"
232 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DEnqueueTPUEmbeddingSparseTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingSparseTensorBatch"
50 name: "EnqueueTPUEmbeddingSparseTensorBatch"
137 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt2 graph_op_name: "EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1356 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1356 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt22338 name: "EnqueueTPUEmbeddingSparseTensorBatch"
22386 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go48796 func EnqueueTPUEmbeddingSparseTensorBatch(scope *Scope, sample_indices []tf.Output, embedding_indic… func
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td3828 def TF_EnqueueTPUEmbeddingSparseTensorBatchOp : TF_Op<"EnqueueTPUEmbeddingSparseTensorBatch", [Same…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt14197 name: "EnqueueTPUEmbeddingSparseTensorBatch"