Home
last modified time | relevance | path

Searched refs:TPUExecute (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu-dynamic-layout-pass.mlir31 // CHECK-NEXT: "tf.TPUExecute"(%[[COPY0]], %[[COPY1]], %[[COMPILE]]#1)
33 %3 = "tf.TPUExecute"(%2#0, %2#1, %compile#1)
64 %3 = "tf.TPUExecute"(%2#0, %2#1, %compile#1)
70 %5 = "tf.TPUExecute"(%4#0, %4#1, %compile#1)
100 %3 = "tf.TPUExecute"(%2#0, %2#1, %compile#1)
131 %3 = "tf.TPUExecute"(%2#0, %2#1, %compile#1)
165 %3 = "tf.TPUExecute"(%2#0, %2#1, %compile#1)
197 %3 = "tf.TPUExecute"(%2#0, %2#1, %compile#1)
226 %3 = "tf.TPUExecute"(%arg0, %2, %compile#1)
268 // CHECK: "tf.TPUExecute"(%[[R0]], %[[R1]], %[[COMPILE]]#1)
[all …]
Dtpu_compile_op_replication_pass.mlir13 …// CHECK-NEXT: %[[EXECUTE_0:.*]] = "tf.TPUExecute"(%[[ADD_0]], %[[PROGRAM_1]]) {device = "/job:wor…
14 …%1 = "tf.TPUExecute"(%0, %program) {device = "/job:worker/replica:0/task:0/device:TPU:0"} : (tenso…
17 …// CHECK-NEXT: %[[EXECUTE_1:.*]] = "tf.TPUExecute"(%[[ADD_1]], %[[PROGRAM_0]]) {device = "/job:wor…
18 …%3 = "tf.TPUExecute"(%2, %program) {device = "/job:worker/replica:0/task:1/device:TPU:0"} : (tenso…
Dtpu-merge-variables-with-execute.mlir37 %0:2 = "tf.TPUExecute"(%read0, %read1, %read2, %compile#1) {
84 %0 = "tf.TPUExecute"(%read0, %read1, %compile#1)
144 %0:3 = "tf.TPUExecute"(%read0, %read1, %read2, %read5, %compile#1) {
180 // CHECK-NEXT: "tf.TPUExecute"(%[[READ_0]], %[[READ_1]], %[[ARG_1]])
182 %0 = "tf.TPUExecute"(%read0, %read1, %arg1) {
209 // CHECK-NEXT: "tf.TPUExecute"(%[[READ_0]], %[[ARG_1]])
211 %0:2 = "tf.TPUExecute"(%read0, %arg1) {
247 %0 = "tf.TPUExecute"(%read0, %arg2) : (tensor<32xf32>, tensor<!tf.string>) -> tensor<32xf32>
257 %1 = "tf.TPUExecute"(%read1, %arg2) : (tensor<64xf32>, tensor<!tf.string>) -> tensor<64xf32>
300 %0 = "tf.TPUExecute"(%read0, %arg4) : (tensor<32xf32>, tensor<!tf.string>) -> tensor<32xf32>
[all …]
Dtpu_rewrite.mlir750 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE_OUTPUT]]#1)
797 // CHECK-NEXT: "tf.TPUExecute"(%[[RI_0]], %[[COMPILE_OUTPUT]]#1)
870 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE_OUTPUT]]#1)
917 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE_OUTPUT]]#1)
964 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE_OUTPUT]]#1)
1016 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE_OUTPUT]]#1)
1060 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE0_OUTPUT]]#1)
1074 // CHECK-NEXT: "tf.TPUExecute"(%[[EXECUTE0_OUTPUT]], %[[COMPILE1_OUTPUT]]#1)
1116 // CHECK-NEXT: "tf.TPUExecute"(%[[A_OUTPUT]], %[[COMPILE0_OUTPUT]]#1)
1130 // CHECK-NEXT: "tf.TPUExecute"(%[[EXECUTE0_OUTPUT]], %[[COMPILE1_OUTPUT]]#1)
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUExecute.pbtxt2 graph_op_name: "TPUExecute"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUExecute.pbtxt2 name: "TPUExecute"
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.h40 xla::StatusOr<xla::ExecutionOutput> TPUExecute(
Dtpu_execute.cc412 xla::StatusOr<xla::ExecutionOutput> TPUExecute( in TPUExecute() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir7 // CHECK: TPUExecute
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_passes.td298 such resource variables can be fused with generated `tf.TPUExecute` ops, which
601 and `tf.TPUExecute` operations. `tf._TPUCompileMlir` contains a MLIR module that is
631 %2 = "tf.TPUExecute"(%arg0, %0#1) : (tensor<i8>, tensor<3x!tf.string>) -> tensor<i8>
664 %3 = "tf.TPUExecute"(%arg2, %1#1) : (tensor<i8>, tensor<3x!tf.string>) -> tensor<i8>
695 %3 = "tf.TPUExecute"(%arg0, %0#1) : (tensor<8xi32>, tensor<3x!tf.string>) -> tensor<8xi32>
701 "tf.TPUExecute"(%0#2) : (tensor<3x!tf.string>) -> ()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc216 TPUExecute(executable, host_transfer_info, in RunExecutable()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc756 TPUExecute(executable, host_transfer_info, in DoWork()
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md2936 ### tf.TPUExecute
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td15715 def TF_TPUExecuteOp : TF_Op<"TPUExecute", [DeclareOpInterfaceMethods<MemoryEffectsOpInterface>]> {
18527 'program' output is a string key that is passed to the TPUExecute op and used to
/external/tensorflow/tensorflow/go/op/
Dwrappers.go46593 func TPUExecute(scope *Scope, args []tf.Output, key tf.Output, Tresults []tf.DataType) (results []t… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53801 name: "TPUExecute"