Searched refs:b_call (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 237 XlaBuilder b_call("add"); in TEST_F() local
238 Parameter(&b_call, 0, ShapeUtil::MakeShape(PRED, {}), "x"); in TEST_F()
251 XlaBuilder b_call("the_only_to_apply"); in TEST_F() local
252 auto p0 = Parameter(&b_call, 0, ShapeUtil::MakeShape(F32, {}), "p0"); in TEST_F()
253 auto p1 = Parameter(&b_call, 1, ShapeUtil::MakeShape(F32, {}), "p1"); in TEST_F()
255 TF_ASSERT_OK_AND_ASSIGN(auto call, b_call.Build()); in TEST_F()
538 XlaBuilder b_call("the_only_to_apply"); in TEST_F() local
539 auto p0 = Parameter(&b_call, 0, ShapeUtil::MakeShape(F32, {}), "p0"); in TEST_F()
540 auto p1 = Parameter(&b_call, 1, ShapeUtil::MakeShape(F32, {}), "p1"); in TEST_F()
542 TF_ASSERT_OK_AND_ASSIGN(auto call, b_call.Build()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1617 auto b_call = b_builder.AddInstruction( in TEST_F() local
1641 GetAllocation(*assignment, b_call, /*index=*/{})); in TEST_F()
1642 EXPECT_EQ(GetAllocation(*assignment, b_call, /*index=*/{}), in TEST_F()
1647 GetAllocation(*assignment, b_call, /*index=*/{0})); in TEST_F()
1648 EXPECT_EQ(GetAllocation(*assignment, b_call, /*index=*/{0}), in TEST_F()
|