Searched refs:MakeBinaryAdd (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | all_gather_decomposer.cc | 38 HloComputation* MakeBinaryAdd(PrimitiveType type, HloModule* module) { in MakeBinaryAdd() function
74 MakeBinaryAdd(dus->shape().element_type(), comp->parent()), in DecomposeAllGather()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 595 MakeBinaryAdd(dot->shape().element_type(), module)); in __anon991ccdc40802()
1881 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionBaseCase()
1987 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionBaseCase()
2696 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionDotGroupOnContracting()
2717 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionDotGroupOnContracting()
2723 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionDotGroupOnContracting()
2779 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionDotGroupOnContracting()
|
| H A D | convolution_handler.cc | 505 b, conv, MakeBinaryAdd(original_hlo->shape().element_type(), module), {}, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
732 b, conv, MakeBinaryAdd(output_base_shape.element_type(), module), {}, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
|
| H A D | spmd_partitioner_util.h | 137 HloComputation* MakeBinaryAdd(PrimitiveType type, HloModule* module);
|
| H A D | gather_scatter_handler.cc | 446 MakeBinaryAdd(PRED, indices.state().module))); in PartitionGatherTrivialIndexedOperandDimension()
468 MakeBinaryAdd(filtered->shape().element_type(), in PartitionGatherTrivialIndexedOperandDimension()
|
| H A D | spmd_partitioner.cc | 1290 MakeBinaryAdd(shard_shape.element_type(), state_.module); in ReplicatePartial()
1474 MakeBinaryAdd(shape.element_type(), state_.module); in Broadcast()
2370 MakeBinaryAdd(hlo->shape().element_type(), module_), in HandleConcatenate()
|
| H A D | spmd_partitioner_util.cc | 57 HloComputation* MakeBinaryAdd(PrimitiveType type, HloModule* module) { in MakeBinaryAdd() function
|