| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dynamic_padder_test.cc | 113 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( local
129 zx_shape, a_param, b_param, /*feature_group_count=*/1,
156 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( local
172 zx_shape, a_param, b_param, /*feature_group_count=*/1,
|
| dynamic_dimension_inference_test.cc | 253 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( local
262 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums,
297 auto* b_param = builder.AddInstruction(HloInstruction::CreateParameter( local
313 zx_shape, a_param, b_param, /*feature_group_count=*/1,
|
| buffer_assignment_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| array_elementwise_ops_test.cc | 331 auto b_param = Parameter(&builder, 1, a_literal.shape(), "b_param"); local
334 auto sum1 = Add(a_constant, b_param);
336 auto sum3 = Add(a_param, b_param);
[all...] |
