Searched refs:b_call (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 225 XlaBuilder b_call("add"); in TEST_F() local
226 Parameter(&b_call, 0, ShapeUtil::MakeShape(PRED, {}), "x"); in TEST_F()
239 XlaBuilder b_call("the_only_to_apply"); in TEST_F() local
240 auto p0 = Parameter(&b_call, 0, ShapeUtil::MakeShape(F32, {}), "p0"); in TEST_F()
241 auto p1 = Parameter(&b_call, 1, ShapeUtil::MakeShape(F32, {}), "p1"); in TEST_F()
243 TF_ASSERT_OK_AND_ASSIGN(auto call, b_call.Build()); in TEST_F()
466 XlaBuilder b_call("the_only_to_apply"); in TEST_F() local
467 auto p0 = Parameter(&b_call, 0, ShapeUtil::MakeShape(F32, {}), "p0"); in TEST_F()
468 auto p1 = Parameter(&b_call, 1, ShapeUtil::MakeShape(F32, {}), "p1"); in TEST_F()
470 TF_ASSERT_OK_AND_ASSIGN(auto call, b_call.Build()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment_test.cc | 1445 auto b_call = b_builder.AddInstruction( in TEST_F() local
1469 GetAllocation(*assignment, b_call, /*index=*/{})); in TEST_F()
1470 EXPECT_EQ(GetAllocation(*assignment, b_call, /*index=*/{}), in TEST_F()
1475 GetAllocation(*assignment, b_call, /*index=*/{0})); in TEST_F()
1476 EXPECT_EQ(GetAllocation(*assignment, b_call, /*index=*/{0}), in TEST_F()
|