Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
97 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dreplicate_to_island.mlir200 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf_type.string>) -…
215 // CHECK: "tf.EnqueueTPUEmbeddingRaggedTensorBatch"
224 // CHECK: "tf.EnqueueTPUEmbeddingRaggedTensorBatch"
H A Dside-effect-analysis-test.mlir1483 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0) {table_ids = [1, 2], device_ordinal = 1} : (tenso…
1486 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0) {table_ids = [1, 2], device_ordinal = 1} : (tenso…
1515 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0) {table_ids = [1, 2], device_ordinal = 1} : (tenso…
1518 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0) {table_ids = [1, 2], device_ordinal = 2} : (tenso…
1546 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0){table_ids = [1, 2]} : (tensor<!tf_type.string>) -…
1553 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0){table_ids = [1, 2]} : (tensor<!tf_type.string>) -…
1584 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0){table_ids = [1, 2], device_ordinal = 1} : (tensor…
1590 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0){table_ids = [1, 2], device_ordinal = 2} : (tensor…
1649 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0){table_ids = [1, 2], device_ordinal = 1} : (tensor…
1655 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0){table_ids = [1, 2], device_ordinal = 5} : (tensor…
[all …]
H A Dtpu_extract_head_tail_outside_compilation.mlir541 // CHECK-NEXT: "tf.EnqueueTPUEmbeddingRaggedTensorBatch"
551 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg0) {_xla_outside_compilation = "cluster1", table_ids…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt2 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt2 graph_op_name: "EnqueueTPUEmbeddingRaggedTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt1428 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt1428 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go14544 func EnqueueTPUEmbeddingRaggedTensorBatch(scope *Scope, sample_splits []tf.Output, embedding_indice… func
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt15660 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td4361 def TF_EnqueueTPUEmbeddingRaggedTensorBatchOp : TF_Op<"EnqueueTPUEmbeddingRaggedTensorBatch", [Decl…