Home
last modified time | relevance | path

Searched refs:EnqueueTPUEmbeddingArbitraryTensorBatch (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dreplica_id_to_device_ordinal.mlir12 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%1){_xla_replica_id = 0 : i64, device_ordinal = -1 : …
17 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%1){_xla_replica_id = 1 : i64, device_ordinal = -1 : …
22 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%1){_xla_replica_id = 2 : i64, device_ordinal = -1 : …
27 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%1){_xla_replica_id = 3 : i64, device_ordinal = -1 : …
38 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
40 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
42 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
44 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
59 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%1){_xla_replica_id = 0 : i64, device_ordinal = -1 : …
69 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
[all …]
H A Dreplicate_to_island.mlir201 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf_type.string>…
217 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
226 // CHECK: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
H A Dtpu_extract_head_tail_outside_compilation.mlir542 // CHECK-NEXT: "tf.EnqueueTPUEmbeddingArbitraryTensorBatch"
552 …"tf.EnqueueTPUEmbeddingArbitraryTensorBatch"(%arg0) {_xla_outside_compilation = "cluster1", table_…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingArbitraryTensorBatch"
85 name: "EnqueueTPUEmbeddingArbitraryTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_EnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt2 graph_op_name: "EnqueueTPUEmbeddingArbitraryTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt1420 name: "EnqueueTPUEmbeddingArbitraryTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt1420 name: "EnqueueTPUEmbeddingArbitraryTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go14324 func EnqueueTPUEmbeddingArbitraryTensorBatch(scope *Scope, sample_indices_or_row_splits []tf.Output… func
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt15517 name: "EnqueueTPUEmbeddingArbitraryTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td4260 def TF_EnqueueTPUEmbeddingArbitraryTensorBatchOp : TF_Op<"EnqueueTPUEmbeddingArbitraryTensorBatch",…