| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| gpu_ldg_test.cc | 118 HloInstruction* reduce = builder.AddInstruction(HloInstruction::CreateReduce(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_dce_test.cc | 342 builder.AddInstruction(HloInstruction::CreateReduce(
383 builder.AddInstruction(HloInstruction::CreateReduce(
393 builder.AddInstruction(HloInstruction::CreateReduce(
|
| batchnorm_expander.cc | 247 auto sum = add(HloInstruction::CreateReduce(feature_shape, operand, zero,
252 auto squared_sum = add(HloInstruction::CreateReduce(
529 add(HloInstruction::CreateReduce(
534 auto grad_beta = add(HloInstruction::CreateReduce(
|
| dynamic_padder_test.cc | 87 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce(
|
| dynamic_dimension_inference_test.cc | 194 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce(
227 HloInstruction::CreateReduce(reduce_shape, negate, init, {1}, GetAdd()));
|
| bfloat16_normalization_test.cc | 204 HloInstruction* reduce = builder.AddInstruction(HloInstruction::CreateReduce(
|
| hlo_creation_utils.cc | 271 return operand->parent()->AddInstruction(HloInstruction::CreateReduce(
|
| hlo_instruction.h | 608 static std::unique_ptr<HloInstruction> CreateReduce(
623 static std::unique_ptr<HloInstruction> CreateReduce(
[all...] |
| algebraic_simplifier.cc | 322 return computation_->AddInstruction(HloInstruction::CreateReduce(
[all...] |
| hlo_evaluator_test.cc | [all...] |
| hlo_instruction.cc | 250 CreateReduce(shape, inputs, init_values,
[all...] |
| algebraic_simplifier_test.cc | 387 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation));
390 builder.AddInstruction(HloInstruction::CreateReduce(r1f32, reduce0, zero,
[all...] |
| hlo_instruction_test.cc | 391 HloInstruction::CreateReduce(f32v100, param0, const0,
[all...] |
| buffer_assignment_test.cc | [all...] |
| hlo_parser.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_fusion_test.cc | 169 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce(
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| fusion_test.cc | 613 auto reduce2 = builder.AddInstruction(HloInstruction::CreateReduce(
633 auto reduce2 = builder.AddInstruction(HloInstruction::CreateReduce(
[all...] |
