Searched refs:collapsed_paddings (Results 1 – 1 of 1) sorted by relevance
109 Tensor collapsed_paddings; in Compute() local113 &collapsed_paddings, &collapsed_output_shape)) { in Compute()123 const Tensor& collapsed_paddings_ref = collapsed_paddings; in Compute()167 std::vector<std::pair<int, int>> collapsed_paddings; in CollapseAdjacentNonPaddedDimensions() local176 collapsed_paddings.push_back({paddings(i, 0), paddings(i, 1)}); in CollapseAdjacentNonPaddedDimensions()195 collapsed_paddings.push_back({0, 0}); in CollapseAdjacentNonPaddedDimensions()202 TensorShape({static_cast<int64>(collapsed_paddings.size()), 2})); in CollapseAdjacentNonPaddedDimensions()205 for (size_t i = 0; i < collapsed_paddings.size(); ++i) { in CollapseAdjacentNonPaddedDimensions()206 collapsed_paddings_as_matrix(i, 0) = collapsed_paddings[i].first; in CollapseAdjacentNonPaddedDimensions()207 collapsed_paddings_as_matrix(i, 1) = collapsed_paddings[i].second; in CollapseAdjacentNonPaddedDimensions()