Searched refs:pad_config (Results 1 – 3 of 3) sorted by relevance
53 PaddingConfig pad_config = MakeNoPaddingConfig(shape.rank()); in PadInstruction() local61 pad_config.mutable_dimensions(dim)->set_edge_padding_high( in PadInstruction()70 HloInstruction::CreatePad(new_shape, instr, zero, pad_config)); in PadInstruction()
3180 const PaddingConfig& pad_config = pad->padding_config(); in HandleReduceWindow() local3181 if (HasInteriorPadding(pad_config)) { in HandleReduceWindow()3223 if (!window_util::HasSymmetricPadding(pad_config)) { in HandleReduceWindow()3227 for (int64 i = 0; i < pad_config.dimensions_size(); ++i) { in HandleReduceWindow()3228 const auto& pad_dimension = pad_config.dimensions(i); in HandleReduceWindow()3239 const auto& pad_dimension = pad_config.dimensions(i); in HandleReduceWindow()3288 TF_RET_CHECK(pad_config.dimensions_size() == rank); in HandleReduceWindow()3291 const auto& pad_dim = pad_config.dimensions(i); in HandleReduceWindow()
1421 const PaddingConfig& pad_config = pad->padding_config(); in HandlePad() local1429 pad_config.dimensions(i).edge_padding_low() + in HandlePad()1430 input_index[i] * (pad_config.dimensions(i).interior_padding() + 1); in HandlePad()