Home
last modified time | relevance | path

Searched refs:CreateBroadcast (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc41 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 …]
Dfusion_test.cc290 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()
Dmultioutput_fusion_test.cc80 HloInstruction::CreateBroadcast(elem_shape2, add1, {})); in RunTest2D()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc107 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 …]
Dconvolution_group_converter.cc183 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()
Ddynamic_padder.cc161 computation->AddInstruction(HloInstruction::CreateBroadcast( in Run()
169 computation->AddInstruction(HloInstruction::CreateBroadcast( in Run()
Dbuffer_assignment_test.cc432 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 …]
Dalgebraic_simplifier.cc814 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 …]
Dhlo_rematerialization_test.cc66 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()
Dmap_inliner.cc94 HloInstruction::CreateBroadcast(map->shape(), constant, {})); in HandleMap()
Dhlo_instruction_test.cc906 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()
Dalgebraic_simplifier_test.cc458 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 …]
Dcopy_insertion_test.cc454 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()
Dtuple_points_to_analysis_test.cc1138 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()
Dlayout_assignment_test.cc437 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()
Dhlo_dataflow_analysis_test.cc2159 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()
Dhlo_creation_utils.cc183 return computation->AddInstruction(HloInstruction::CreateBroadcast( in MakeBroadcastHlo()
Dhlo_cost_analysis_test.cc510 HloInstruction::CreateBroadcast(shape_without_layout, c2, {1})); in TEST_F()
Dheap_simulator_test.cc177 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/
Dcudnn_batchnorm_rewriter.cc132 computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormTraining()
141 computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormTraining()
188 computation_->AddInstruction(HloInstruction::CreateBroadcast( in HandleBatchNormGrad()
Dwhile_transformer_test.cc75 HloInstruction::CreateBroadcast(data_shape_, cast, {})); in BuildBodyComputation()
Dinstruction_fusion_test.cc41 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F()
61 builder.AddInstruction(HloInstruction::CreateBroadcast( in TEST_F()
Dcudnn_fused_conv_rewriter.cc212 bias = computation->AddInstruction(HloInstruction::CreateBroadcast( in TryRewriteToCudnnForwardRelu()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc106 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/
Dcpu_instruction_fusion_test.cc346 HloInstruction::CreateBroadcast(broadcast_shape, param0, {1})); in TEST_F()
372 HloInstruction::CreateBroadcast(result_shape, param0, {1})); in TEST_F()

12