Home
last modified time | relevance | path

Searched refs:send0 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver_test.cc159 XlaOp send0 = outfeed_receiver in TEST() local
163 EXPECT_TRUE(CompileAndExecute(&builder0, send0, 0, cpu_client.get()).ok()); in TEST()
204 XlaOp send0 = outfeed_receiver in TEST() local
214 ->AddOutfeedToBuilder(&builder, send0, consumer_id1, {data1}) in TEST()
247 XlaOp send0 = outfeed_receiver in TEST() local
256 &builder, send0, consumer_id0, {data1}); in TEST()
280 StatusOr<XlaOp> send0 = outfeed_receiver->AddOutfeedToBuilder( in TEST() local
283 EXPECT_FALSE(send0.ok()); in TEST()
284 EXPECT_THAT(send0.status().ToString(), in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding_test.cc461 auto send0 = ops::_Send(s.WithOpName("send0"), add0, "send0", "sender", 0, in TEST_F() local
490 Node* send0 = index.at("send0"); in TEST_F() local
494 ASSERT_EQ(1, send0->num_inputs()); in TEST_F()
495 Node* cf0 = *(send0->in_nodes().begin()); in TEST_F()
538 auto send0 = ops::_Send(s.WithOpName("send0"), rank0, "send0", "sender", 0, in TEST_F() local
569 Node* send0 = index.at("send0"); in TEST_F() local
574 ASSERT_EQ(1, send0->num_inputs()); in TEST_F()
575 Node* cf0 = *(send0->in_nodes().begin()); in TEST_F()
601 auto send0 = ops::_Send(s.WithOpName("send0"), add0, "send0", "sender", 0, in TEST_F() local
621 Node* send0 = index.at("send0"); in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-communication.mlir229 …%0 = "tf._XlaHostComputeMlir"(%arg0) {recv_key = "recv0", send_key = "send0", tpu_core = 0 : i64} …
344 "tf.XlaSendToHost"(%arg0) {key = "send0"} : (tensor<i32>) -> ()