Searched refs:send0 (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | outfeed_receiver_test.cc | 159 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/ |
D | constant_folding_test.cc | 461 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/ |
D | legalize-tf-communication.mlir | 229 …%0 = "tf._XlaHostComputeMlir"(%arg0) {recv_key = "recv0", send_key = "send0", tpu_core = 0 : i64} … 344 "tf.XlaSendToHost"(%arg0) {key = "send0"} : (tensor<i32>) -> ()
|