/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | broadcast_test.cc | 41 builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 57 builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 77 auto element1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 79 auto element2 = builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 101 builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 120 builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 137 builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() 157 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 182 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 213 builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F() [all …]
|
D | fusion_test.cc | 290 builder.AddInstruction(HloInstruction::CreateBroadcast(shape, two, {})); in XLA_TEST_F() 315 HloInstruction::CreateBroadcast(const_array->shape(), const_vector, {1})); in XLA_TEST_F() 505 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in XLA_TEST_F()
|
D | multioutput_fusion_test.cc | 80 HloInstruction::CreateBroadcast(elem_shape2, add1, {})); in RunTest2D()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | batchnorm_expander.cc | 107 HloInstruction::CreateBroadcast(operand->shape(), element_count, {})); in Mean() 220 auto epsilon = add(HloInstruction::CreateBroadcast( in HandleBatchNormTraining() 235 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index})); in HandleBatchNormTraining() 238 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index})); in HandleBatchNormTraining() 260 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index})); in HandleBatchNormTraining() 274 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index})); in HandleBatchNormTraining() 345 auto epsilon = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormInference() 373 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index})); in HandleBatchNormInference() 376 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index})); in HandleBatchNormInference() 379 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index})); in HandleBatchNormInference() [all …]
|
D | convolution_group_converter.cc | 183 auto broadcasted_mask1 = add_instruction(HloInstruction::CreateBroadcast( in GetExpandedFilterMask() 187 auto broadcasted_mask2 = add_instruction(HloInstruction::CreateBroadcast( in GetExpandedFilterMask() 248 add(HloInstruction::CreateBroadcast(expanded_filter_shape, zero, {})); in HandleBatchGroupCount() 369 expanded_filter = add(HloInstruction::CreateBroadcast( in HandleConvolution() 375 add(HloInstruction::CreateBroadcast(expanded_filter_shape, zero, {})); in HandleConvolution()
|
D | dynamic_padder.cc | 161 computation->AddInstruction(HloInstruction::CreateBroadcast( in Run() 169 computation->AddInstruction(HloInstruction::CreateBroadcast( in Run()
|
D | buffer_assignment_test.cc | 432 HloInstruction::CreateBroadcast(f32vec100_, paramscalar, {})); in TEST_F() 515 HloInstruction::CreateBroadcast(f32vec100_, paramscalar, {})); in TEST_F() 565 HloInstruction::CreateBroadcast(f32vec100_, paramscalar, {})); in TEST_F() 626 HloInstruction::CreateBroadcast(f32vec100_, paramscalar, {})); in TEST_F() 698 HloInstruction::CreateBroadcast(f32vec100_, paramscalar, {})); in TEST_F() 1006 HloInstruction::CreateBroadcast(f32a100x10_, slice, {1})); in TEST_F() 1038 HloInstruction::CreateBroadcast(f32a100x10_, slice, {1})); in TEST_F() 1075 HloInstruction::CreateBroadcast(f32a100x10_, slice, {1})); in TEST_F() 1110 auto broadcast = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 1142 auto broadcast = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() [all …]
|
D | algebraic_simplifier.cc | 814 HloInstruction::CreateBroadcast(constant->shape(), scalar, {})); in HandleConstant() 895 HloInstruction::CreateBroadcast(b->shape(), shift_amount, {})); in TryDivideToShift() 919 HloInstruction::CreateBroadcast(b->shape(), shift_amount, {})); in TryDivideToShift() 1193 HloInstruction::CreateBroadcast(shape, hlo, dims)); in HandleDotStrengthReduction() 1644 dot, HloInstruction::CreateBroadcast(dot->shape(), zero, {})); in HandleDot() 1667 new_lhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot() 1684 new_rhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot() 1734 new_lhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot() 1746 new_rhs = computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleDot() 2077 HloInstruction::CreateBroadcast(broadcast->shape(), in HandleBroadcast() [all …]
|
D | hlo_rematerialization_test.cc | 66 HloInstruction::CreateBroadcast(vec1024_shape_, reshape, {})); in MakeRematerializableComputation() 111 HloInstruction::CreateBroadcast(vec1024_shape_, reshape, {})); in MakeRematerializableWhileComputation() 443 HloInstruction::CreateBroadcast(vec1024_shape_, param, {})); in TEST_F() 602 HloInstruction::CreateBroadcast(vec1024_shape_, param, {})); in TEST_P()
|
D | map_inliner.cc | 94 HloInstruction::CreateBroadcast(map->shape(), constant, {})); in HandleMap()
|
D | hlo_instruction_test.cc | 906 EXPECT_TRUE(Identical(*HloInstruction::CreateBroadcast(shape, op1, {0, 1}), in TEST_F() 907 *HloInstruction::CreateBroadcast(shape, op1, {0, 1}))); in TEST_F() 908 EXPECT_FALSE(Identical(*HloInstruction::CreateBroadcast(shape, op1, {0, 1}), in TEST_F() 909 *HloInstruction::CreateBroadcast(shape, op1, {1, 0}))); in TEST_F() 913 Identical(*HloInstruction::CreateBroadcast(bcast_shape1, op1, {0, 1}), in TEST_F() 914 *HloInstruction::CreateBroadcast(bcast_shape2, op1, {0, 1}))); in TEST_F() 1067 builder.AddInstruction(HloInstruction::CreateBroadcast(r2f32, p3, {1})); in TEST_F() 1106 builder.AddInstruction(HloInstruction::CreateBroadcast(r1f32, y, {})); in TEST_F() 1249 HloInstruction::CreateBroadcast(data_shape, one, {})); in TEST_F()
|
D | algebraic_simplifier_test.cc | 458 HloInstruction::CreateBroadcast(r2f32, zero, {0, 1})); in TEST_F() 495 HloInstruction::CreateBroadcast(r2f32, zero, {}))}, in TEST_F() 541 builder.AddInstruction(HloInstruction::CreateBroadcast(r2f32, zero, {1})); in TEST_F() 1506 auto broadcast = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 1796 HloInstruction::CreateBroadcast(r1f32, param1, {})); in TEST_F() 2092 HloInstruction::CreateBroadcast(ShapeUtil::MakeShape(F32, {2, 2, 2}), add, in TEST_F() 2272 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 2293 auto broadcast1 = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 2315 auto broadcast = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 2337 auto broadcast = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() [all …]
|
D | copy_insertion_test.cc | 454 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8})); in BuildDependentBodyComputation() 524 HloInstruction::CreateBroadcast(data_shape_, induction_variable, {8})); in BuildDependentBodyOneReadOnlyComputation() 687 HloInstruction::CreateBroadcast(data_shape_, one, {1})); in BuildWhileInstruction_InitPointsToAmbiguous() 691 HloInstruction::CreateBroadcast(data_shape_, zero, {1})); in BuildWhileInstruction_InitPointsToAmbiguous() 711 HloInstruction::CreateBroadcast(data_shape_, one, {1})); in BuildWhileInstruction_InitPointsToNonDistinct() 724 HloInstruction::CreateBroadcast(data_shape_, one, {1})); in BuildWhileInstruction_InitPointsToInterfering()
|
D | tuple_points_to_analysis_test.cc | 1138 HloInstruction::CreateBroadcast(data_shape, one, {1})); in TEST_F() 1160 HloInstruction::CreateBroadcast(data_shape, one, {1})); in TEST_F() 1233 HloInstruction::CreateBroadcast(shape, one, {1})); in TEST_F() 1269 HloInstruction::CreateBroadcast(full_shape, param1, {0})); in TEST_F()
|
D | layout_assignment_test.cc | 437 HloInstruction::CreateBroadcast(bshape, param, {1, 2})); in TEST_F() 480 HloInstruction::CreateBroadcast(f32_34, param, {1})); in TEST_F() 486 HloInstruction::CreateBroadcast(f32_234, tanh, {1, 2})); in TEST_F() 606 HloInstruction::CreateBroadcast(input_shape, constant, {})); in TEST_F()
|
D | hlo_dataflow_analysis_test.cc | 2159 HloInstruction::CreateBroadcast(data_shape, one, {1})); in TEST_F() 2517 HloInstruction::CreateBroadcast(data_shape, one, {1})); in TEST_F() 2539 HloInstruction::CreateBroadcast(data_shape, one, {1})); in TEST_F() 2567 HloInstruction::CreateBroadcast(data_shape, one, {1})); in TEST_F() 2639 HloInstruction::CreateBroadcast(shape, one, {1})); in TEST_F()
|
D | hlo_creation_utils.cc | 183 return computation->AddInstruction(HloInstruction::CreateBroadcast( in MakeBroadcastHlo()
|
D | hlo_cost_analysis_test.cc | 510 HloInstruction::CreateBroadcast(shape_without_layout, c2, {1})); in TEST_F()
|
D | heap_simulator_test.cc | 177 HloInstruction::CreateBroadcast(r2f32, while_loop, {1})); in TEST_F() 656 HloInstruction::CreateBroadcast(f32vec4_, element0, {0})); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_batchnorm_rewriter.cc | 132 computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormTraining() 141 computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormTraining() 188 computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormGrad()
|
D | while_transformer_test.cc | 75 HloInstruction::CreateBroadcast(data_shape_, cast, {})); in BuildBodyComputation()
|
D | instruction_fusion_test.cc | 41 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 61 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F()
|
D | cudnn_fused_conv_rewriter.cc | 212 bias = computation->AddInstruction(HloInstruction::CreateBroadcast( in TryRewriteToCudnnForwardRelu()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_fusion_test.cc | 106 auto two = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F() 182 auto two = builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_instruction_fusion_test.cc | 346 HloInstruction::CreateBroadcast(broadcast_shape, param0, {1})); in TEST_F() 372 HloInstruction::CreateBroadcast(result_shape, param0, {1})); in TEST_F()
|