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 D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 2 name: "EnqueueTPUEmbeddingRaggedTensorBatch" 97 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
H A D | replicate_to_island.mlir | 200 …"tf.EnqueueTPUEmbeddingRaggedTensorBatch"(%arg2){table_ids = [1, 2]} : (tensor<!tf_type.string>) -… 215 // CHECK: "tf.EnqueueTPUEmbeddingRaggedTensorBatch" 224 // CHECK: "tf.EnqueueTPUEmbeddingRaggedTensorBatch"
|
H A D | side-effect-analysis-test.mlir | 1483 …"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 D | tpu_extract_head_tail_outside_compilation.mlir | 541 // 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 D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 2 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
H A D | api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 2 graph_op_name: "EnqueueTPUEmbeddingRaggedTensorBatch"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.raw_ops.pbtxt | 1428 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.raw_ops.pbtxt | 1428 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
|
/aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
H A D | wrappers.go | 14544 func EnqueueTPUEmbeddingRaggedTensorBatch(scope *Scope, sample_splits []tf.Output, embedding_indice… func
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | ops.pbtxt | 15660 name: "EnqueueTPUEmbeddingRaggedTensorBatch"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_generated_ops.td | 4361 def TF_EnqueueTPUEmbeddingRaggedTensorBatchOp : TF_Op<"EnqueueTPUEmbeddingRaggedTensorBatch", [Decl…
|