Home
last modified time | relevance | path

Searched refs:MakeBroadcastHlo (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.h103 HloInstruction* MakeBroadcastHlo(HloInstruction* operand,
106 HloInstruction* MakeBroadcastHlo(HloInstruction* operand,
Dhlo_creation_utils.cc205 HloInstruction* MakeBroadcastHlo(HloInstruction* operand, in MakeBroadcastHlo() function
216 HloInstruction* MakeBroadcastHlo(HloInstruction* operand, in MakeBroadcastHlo() function
219 return MakeBroadcastHlo(operand, broadcast_dimensions, shape.dimensions()); in MakeBroadcastHlo()
554 return MakeBroadcastHlo(zero, /*broadcast_dimensions=*/{}, in BroadcastZeros()
563 return MakeBroadcastHlo(one, /*broadcast_dimensions=*/{}, in BroadcastOnes()
Dhlo_creation_utils_test.cc290 HloInstruction* output = MakeBroadcastHlo(input, {}, {2, 2}); in TEST_F()
310 MakeBroadcastHlo(input, {}, ShapeUtil::MakeShape(S32, {2, 2})); in TEST_F()
Dscatter_expander.cc215 return MakeBroadcastHlo(valid_index_reduced, {}, window_sizes); in CheckIndexValidity()
233 MakeBroadcastHlo(induction_var, /*broadcast_dimensions=*/{}, in ScatterLoopBody()
Dgather_expander.cc165 MakeBroadcastHlo(induction_var, /*broadcast_dimensions=*/{}, in GatherLoopBody()
Dspace_to_batch_converter.cc1038 auto new_broadcast = MakeBroadcastHlo(consumer->mutable_operand(0), in PropagateOnBroadcast()
1526 MakeBroadcastHlo(init_val, {}, top->shape().dimensions()); in Propagate()
1599 HloInstruction* shape_mask = MakeBroadcastHlo( in Propagate()
1693 MakeBroadcastHlo(slice_mask_reshaped, {new_batch_dim, new_space_dim}, in SelectValidPortion()
1699 MakeBroadcastHlo(select_val, {}, new_instr->shape().dimensions()); in SelectValidPortion()
Dalgebraic_simplifier.cc1365 MakeBroadcastHlo(MakeReshapeHlo(new_shape, operands[0]).ValueOrDie(), in HandleConcatenate()
2525 MakeBroadcastHlo(new_operand, {}, gather->shape()); in HandleGather()
4184 MakeBroadcastHlo(pad->mutable_operand(1), {}, slice->shape()); in HandleSlice()