Searched refs:padding_dimensions (Results 1 – 2 of 2) sorted by relevance
3354 std::vector<int64> padding_dimensions; in HandlePad() local3364 padding_dimensions.push_back(dimension_index); in HandlePad()3366 padding_dimensions.push_back(dimension_index); in HandlePad()3394 auto pad_dims = padding_dimensions.size(); in HandlePad()3408 if (broadcast_index == padding_dimensions[i]) { in HandlePad()3425 for (int i = padding_dimensions.size() - 1; i >= 0; --i) { in HandlePad()3426 int64 j = padding_dimensions[i]; in HandlePad()3436 for (auto dimension_to_pad : padding_dimensions) { in HandlePad()3448 HloInstruction::CreateBroadcast(pad_shape, pad2, padding_dimensions)); in HandlePad()
636 std::initializer_list<std::array<int64, 3>> padding_dimensions) { in CreatePaddingConfig() argument639 for (auto& paddings_per_dim : padding_dimensions) { in CreatePaddingConfig()