| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 247 auto sum = add(HloInstruction::CreateReduce(feature_shape, operand, zero, in HandleBatchNormTraining()
252 auto squared_sum = add(HloInstruction::CreateReduce( in HandleBatchNormTraining()
529 add(HloInstruction::CreateReduce( in HandleBatchNormGrad()
534 auto grad_beta = add(HloInstruction::CreateReduce( in HandleBatchNormGrad()
|
| D | hlo_dce_test.cc | 342 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
383 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
393 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | dynamic_padder_test.cc | 87 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 194 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
227 HloInstruction::CreateReduce(reduce_shape, negate, init, {1}, GetAdd())); in TEST_F()
|
| D | hlo_instruction.h | 608 static std::unique_ptr<HloInstruction> CreateReduce(
623 static std::unique_ptr<HloInstruction> CreateReduce(
|
| D | bfloat16_normalization_test.cc | 204 HloInstruction* reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | algebraic_simplifier.cc | 322 return computation_->AddInstruction(HloInstruction::CreateReduce( in AddReduce()
2980 reduce, HloInstruction::CreateReduce( in HandleReduce()
3010 reduce, HloInstruction::CreateReduce( in HandleReduce()
3054 reduce, HloInstruction::CreateReduce( in HandleReduce()
3070 HloInstruction::CreateReduce(reduce_result_shape, operand, init_value, in HandleReduce()
3140 computation_->AddInstruction(HloInstruction::CreateReduce( in HandleReduceWindow()
|
| D | hlo_creation_utils.cc | 271 return operand->parent()->AddInstruction(HloInstruction::CreateReduce( in MakeReduceHlo()
|
| D | hlo_evaluator_test.cc | 1452 HloInstruction::CreateReduce(scalar_shape, arg_instruction, init_value, in TEST_F()
1488 HloInstruction::CreateReduce(scalar_shape, arg_instruction, init_value, in BM_ReducePrecisely()
1530 HloInstruction::CreateReduce(shape, arg_instruction, init_value, in TEST_P()
|
| D | hlo_instruction.cc | 250 CreateReduce(shape, inputs, init_values, in CreateFromProto()
1032 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReduce( in CreateReduce() function in xla::HloInstruction
1041 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReduce( in CreateReduce() function in xla::HloInstruction
|
| D | hlo_instruction_test.cc | 391 HloInstruction::CreateReduce(f32v100, param0, const0, in TEST_F()
|
| D | algebraic_simplifier_test.cc | 387 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation)); in TEST_F()
390 builder.AddInstruction(HloInstruction::CreateReduce(r1f32, reduce0, zero, in TEST_F()
1741 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | buffer_assignment_test.cc | 812 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | hlo_parser.cc | 1209 instruction = builder->AddInstruction(HloInstruction::CreateReduce( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ldg_test.cc | 118 HloInstruction* reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 169 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 613 auto reduce2 = builder.AddInstruction(HloInstruction::CreateReduce( in XLA_TEST_F()
633 auto reduce2 = builder.AddInstruction(HloInstruction::CreateReduce( in XLA_TEST_F()
|