Searched defs:halo_size (Results 1 – 6 of 6) sorted by relevance
218 int halo_size; in HandleConv() local
283 int64_t halo_size = in HandleCustomCallSPMDInternal_RotateRight() local
392 int64_t halo_size = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
762 int64_t halo_size = halo_size_including_skips - halo_right_skips; in ExchangeHalo() local824 int64_t halo_size = halo_size_including_skips - halo_left_skips; in ExchangeHalo() local
432 StatusOr<mlir::Value> EmitHaloExchange(mlir::OpBuilder& builder, int halo_size, in EmitHaloExchange()
74 base_dilation_factor, halo_size, high_padding_for_conv, member512 int64_t activations_batch_dim, int64_t low_padding, int64_t halo_size, in HaloDuplicateWithSlice()2069 const int64_t halo_size = last_overlap_point + window_size - new_space_size; in Propagate() local3596 const int64_t halo_size = is_base_dilated in GetConvolutionDetails() local