| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dynamic_padder_test.cc | 111 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
129 zx_shape, a_param, b_param, /*feature_group_count=*/1,
154 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
172 zx_shape, a_param, b_param, /*feature_group_count=*/1,
|
| dynamic_dimension_inference_test.cc | 251 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
262 HloInstruction::CreateDot(xz_shape, a_param, b_param, dot_dnums,
295 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
313 zx_shape, a_param, b_param, /*feature_group_count=*/1,
341 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
351 HloInstruction::CreateTranspose(output_shape, a_param, {2, 1, 0}));
380 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
386 HloInstruction::CreateReshape(output_shape, a_param));
414 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
420 builder.AddInstruction(HloInstruction::CreateReshape(output_shape, a_param));
439 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
500 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
577 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
627 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
[all...] |
| heap_simulator_test.cc | 451 auto* a_param = builder.AddInstruction(HloInstruction::CreateParameter( local
454 HloInstruction::CreateUnary(f32vec4_, HloOpcode::kExp, a_param));
456 HloInstruction::CreateUnary(f32vec4_, HloOpcode::kNegate, a_param));
462 auto a_param = builder.AddInstruction( local
465 HloInstruction::CreateUnary(f32vec4_, HloOpcode::kNegate, a_param));
471 tracker.RunWholeModule({a_param, neg, fusion});
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| array_elementwise_ops_test.cc | 326 auto a_param = Parameter(&builder, 0, a_literal.shape(), "a_param"); local
336 auto sum3 = Add(a_param, b_param);
337 auto sum4 = Add(a_param, b_constant);
[all...] |
