Searched refs:will_resize (Results 1 – 3 of 3) sorted by relevance
257 if (operands_[arg].stride_bytes.empty() || operands_[arg].will_resize) { in reorder_dimensions()576 if (!op.tensor_base().defined() || op.will_resize) { in allocate_or_resize_outputs()1199 operands_[i].will_resize = true; in mark_resize_outputs()1203 operands_[i].will_resize = true; in mark_resize_outputs()1272 if (op.tensor_base().defined() && !op.will_resize) { in compute_strides()1429 if (op.tensor_base().defined() && !op.will_resize) { in compute_fast_setup_type()1443 if (op.tensor_base().defined() && !op.will_resize) { in compute_fast_setup_type()1462 if (op.tensor_base().defined() && !op.will_resize) { in compute_fast_setup_type()1574 } else if (op.will_resize) { in set_output_raw_strided()1644 } else if (op.will_resize) { in set_output_raw_strided()
181 bool will_resize = false; member
94 TORCH_CHECK(output_operand.will_resize, in abs_kernel()101 TORCH_CHECK(!output_operand.will_resize, "output operand will_resize is True."); in abs_kernel()