Searched refs:pad_cols (Results 1 – 2 of 2) sorted by relevance
436 std::set<int32_t> pad_cols; in PadColumns() local439 RETURN_IF_NOT_OK(UnpackPadInfo(pad_info, column_name_id_map, &pad_cols, &pad_vals, &pad_shapes)); in PadColumns()442 for (size_t col_id : pad_cols) { in PadColumns()453 for (size_t col_id : pad_cols) { // iterator each tensor in a row in PadColumns()465 for (size_t col_id : pad_cols) { in PadColumns()473 for (size_t col_id : pad_cols) { in PadColumns()484 … std::set<int32_t> *pad_cols, std::vector<std::shared_ptr<Tensor>> *pad_vals, in UnpackPadInfo() argument486 RETURN_UNEXPECTED_IF_NULL(pad_cols); in UnpackPadInfo()491 pad_cols->insert(col_id); in UnpackPadInfo()503 pad_cols->insert(col_id); in UnpackPadInfo()
240 … std::set<int32_t> *pad_cols, std::vector<std::shared_ptr<Tensor>> *pad_vals,