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()519 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()745 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionTiledOutput()821 const Window& conv_window)>& create_sharded_conv, in PartitionConvolutionBaseCase()969 const Window& conv_window)>& create_sharded_conv, in PartitionConvolution()1025 auto create_sharded_conv = in HandleConvolution() local