Searched defs:create_sharded_conv (Results 1 – 1 of 1) sorted by relevance
46 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithBatchGroupCount()138 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithFeatureGroupCount()232 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()521 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()747 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionTiledOutput()823 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionBaseCase()971 const Window& conv_window)>& create_sharded_conv, in PartitionConvolution()1030 auto create_sharded_conv = in HandleConvolution() local