Searched refs:MakeBinaryAdd (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_gather_decomposer.cc | 36 HloComputation* MakeBinaryAdd(PrimitiveType type, HloModule* module) { in MakeBinaryAdd() function
131 MakeBinaryAdd(dus->shape().element_type(), comp->parent()), in DecomposeAllGather()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 80 HloComputation* MakeBinaryAdd(PrimitiveType type, HloModule* module);
|
| D | gather_scatter_handler.cc | 270 MakeBinaryAdd(PRED, indices.state().module))); in ParititonTrivialIndexedOperandDimension()
298 MakeBinaryAdd(filtered->shape().element_type(), in ParititonTrivialIndexedOperandDimension()
|
| D | convolution_handler.cc | 503 b, conv, MakeBinaryAdd(original_hlo->shape().element_type(), module), {}, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
730 b, conv, MakeBinaryAdd(output_base_shape.element_type(), module), {}, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
|
| D | dot_handler.cc | 1521 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionBaseCase()
1631 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionBaseCase()
2240 MakeBinaryAdd(output_base_shape.element_type(), module)); in PartitionDotGroupOnContracting()
|
| D | spmd_partitioner.cc | 894 MakeBinaryAdd(shard_shape.element_type(), state_.module); in ReplicatePartial()
1078 MakeBinaryAdd(shape.element_type(), state_.module); in Broadcast()
1564 MakeBinaryAdd(hlo->shape().element_type(), module_), in HandleConcatenate()
|
| D | spmd_partitioner_util.cc | 110 HloComputation* MakeBinaryAdd(PrimitiveType type, HloModule* module) { in MakeBinaryAdd() function
|