Home
last modified time | relevance | path

Searched refs:b_param (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference_test.cc317 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
326 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums, in TEST_F()
358 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
371 HloInstruction::CreateDot(output_shape, a_param, b_param, dot_dnums, in TEST_F()
397 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
408 HloInstruction::CreateDot(output_shape, a_param, b_param, dot_dnums, in TEST_F()
447 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
463 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
Ddynamic_padder_test.cc269 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
285 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
312 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( in TEST_F() local
328 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
Dbuffer_assignment_test.cc1615 auto b_param = b_builder.AddInstruction( in TEST_F() local
1618 HloInstruction::CreateCall(tuple_shape, {b_param}, c_computation.get())); in TEST_F()
1653 EXPECT_TRUE(BuffersDistinct({a_param}, {b_param}, *assignment)); in TEST_F()
1657 EXPECT_EQ(GetAllocation(*assignment, b_param, /*index=*/{0}), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc364 auto b_param = Parameter(&builder, 1, a_literal.shape(), "b_param"); in TEST_P() local
367 auto sum1 = Add(a_constant, b_param); in TEST_P()
369 auto sum3 = Add(a_param, b_param); in TEST_P()