Searched refs:is_variable_input (Results 1 – 6 of 6) sorted by relevance
155 auto& is_variable_input = py_fn->is_variable_input; in apply() local157 auto num_forward_inputs = static_cast<Py_ssize_t>(is_variable_input.size()); in apply()183 return to_variable_list(r.get(), is_variable_input); in apply()196 const auto& is_variable_input = py_fn->is_variable_input; in defer_to_dynamo() local199 TORCH_INTERNAL_ASSERT(is_variable_input.size() >= input_infos.size()); in defer_to_dynamo()204 PyTuple_New(static_cast<Py_ssize_t>(is_variable_input.size()))); in defer_to_dynamo()209 for (const auto i : c10::irange(is_variable_input.size())) { in defer_to_dynamo()210 if (!is_variable_input[i]) { in defer_to_dynamo()254 return to_variable_list(r.get(), is_variable_input); in defer_to_dynamo()335 args.collect(f->is_variable_input); in compiled_args()[all …]
35 const std::vector<bool>& is_variable_input);135 std::vector<bool> is_variable_input; member
78 value->tensor.is_variable_input = tflite_tensor->is_variable; in ReadNonConstantTensor()93 value->tensor.is_variable_input = tflite_tensor->is_variable; in ReadNonConstantTensor()193 if (!value->tensor.is_variable_input) { in AddUpdate()
108 bool is_variable_input = false;
51 [](const ValueDef& v) { return v.value->tensor.is_variable_input; }); in variable_inputs()
3251 if (value->tensor.is_variable_input) { in CopyVariableTensorOutputs()