Home
last modified time | relevance | path

Searched refs:XlaRecvFromHost (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DXlaRecvFromHost.pbtxt2 name: "XlaRecvFromHost"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_XlaRecvFromHost.pbtxt2 graph_op_name: "XlaRecvFromHost"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-communication.mlir138 %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf.shape<>} : () -> tensor<i32>
180 // Test legalization of `tf.XlaRecvFromHost` expands into a `mhlo.recv` op.
200 %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf.shape<>} : () -> tensor<i32>
402 %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf.shape<>} : () -> tensor<i32>
442 %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf.shape<>} : () -> tensor<i32>
1105 %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf.shape<>} : () -> tensor<i32>
/external/tensorflow/tensorflow/go/op/
Dwrappers.go22888 func XlaRecvFromHost(scope *Scope, Toutput tf.DataType, shape tf.Shape, key string) (output tf.Outp… func
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td17893 def TF_XlaRecvFromHostOp : TF_Op<"XlaRecvFromHost", []> {
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt58675 name: "XlaRecvFromHost"