Searched refs:pad_tensor (Results 1 – 3 of 3) sorted by relevance
61 mindspore::MSTensor &pad_tensor = in_tensors_[1]; in AddInnerOp() local62 …int element_cnt = std::accumulate(pad_tensor.Shape().begin(), pad_tensor.Shape().end(), 1, std::mu… in AddInnerOp()100 << ", pad_tensor ElementNum: " << pad_tensor.ElementNum(); in AddInnerOp()
474 std::shared_ptr<Tensor> pad_tensor; in PadColumns() local475 RETURN_IF_NOT_OK(PadEnd(row[col_id], &pad_tensor, pad_shapes[col_id], pad_vals[col_id])); in PadColumns()476 row[col_id] = pad_tensor; in PadColumns()
2368 pad_tensor = _to_tensor(pad_values)2369 pad_shape = pad_tensor.shape