Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A 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…
H A Dreplicate_to_island.mlir199 …"tf.EnqueueTPUEmbeddingSparseTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf_type.string>) -…
213 // CHECK: "tf.EnqueueTPUEmbeddingSparseTensorBatch"
222 // CHECK: "tf.EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DEnqueueTPUEmbeddingSparseTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingSparseTensorBatch"
50 name: "EnqueueTPUEmbeddingSparseTensorBatch"
137 name: "EnqueueTPUEmbeddingSparseTensorBatch"
232 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DEnqueueTPUEmbeddingSparseTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingSparseTensorBatch"
50 name: "EnqueueTPUEmbeddingSparseTensorBatch"
137 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt2 graph_op_name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt1436 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt1436 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v2.pbtxt22338 name: "EnqueueTPUEmbeddingSparseTensorBatch"
22386 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go14728 func EnqueueTPUEmbeddingSparseTensorBatch(scope *Scope, sample_indices []tf.Output, embedding_indic… func
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt15846 name: "EnqueueTPUEmbeddingSparseTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td4449 def TF_EnqueueTPUEmbeddingSparseTensorBatchOp : TF_Op<"EnqueueTPUEmbeddingSparseTensorBatch", [Decl…