Home
last modified time | relevance | path

Searched refs:rhs_shard_size_dilated (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc676 int64 rhs_shard_size_dilated = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
695 wd.stride() * (window_count - 1) + rhs_shard_size_dilated; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()