Searched refs:b_param (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder_test.cc | 113 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
129 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
156 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
172 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 253 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
262 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums, in TEST_F()
297 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
313 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
|
| D | buffer_assignment_test.cc | 1443 auto b_param = b_builder.AddInstruction( in TEST_F() local
1446 HloInstruction::CreateCall(tuple_shape, {b_param}, c_computation.get())); in TEST_F()
1481 EXPECT_TRUE(BuffersDistinct({a_param}, {b_param}, *assignment)); in TEST_F()
1485 EXPECT_EQ(GetAllocation(*assignment, b_param, /*index=*/{0}), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 331 auto b_param = Parameter(&builder, 1, a_literal.shape(), "b_param"); in TEST_P() local
334 auto sum1 = Add(a_constant, b_param); in TEST_P()
336 auto sum3 = Add(a_param, b_param); in TEST_P()
|