Searched refs:excluded_edges (Results 1 – 1 of 1) sorted by relevance
39 int64 excluded_edges = mode == MirrorPadMode::REFLECT ? 1 : 0; in DoMirrorPad() local50 lhs_padding <= dim_size - excluded_edges); in DoMirrorPad()52 rhs_padding <= dim_size - excluded_edges); in DoMirrorPad()55 xla::SliceInDim(t_rev, dim_size - excluded_edges - lhs_padding, in DoMirrorPad()56 dim_size - excluded_edges, 1, dimno); in DoMirrorPad()57 auto rhs_pad = xla::SliceInDim(t_rev, excluded_edges, in DoMirrorPad()58 excluded_edges + rhs_padding, 1, dimno); in DoMirrorPad()