Searched refs:create_sharded_conv (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 46 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithBatchGroupCount() argument
125 create_sharded_conv(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionConvolutionWithBatchGroupCount()
138 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithFeatureGroupCount() argument
217 create_sharded_conv(lhs.hlo(), rhs.hlo(), b, conv_window)); in PartitionConvolutionWithFeatureGroupCount()
232 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() argument
500 auto conv, create_sharded_conv(conv_lhs, rhs_with_halo, b, new_window)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
519 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() argument
727 auto conv, create_sharded_conv(lhs_with_halo, rhs.hlo(), b, new_window)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
745 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionTiledOutput() argument
800 create_sharded_conv(resharded_operand_and_window->sharded_input, in PartitionConvolutionTiledOutput()
[all …]
|
| D | convolution_handler.h | 34 const Window& conv_window)>& create_sharded_conv,
|