Searched refs:new_tensors (Results 1 – 9 of 9) sorted by relevance
28 new_tensors.push_back(desc); in AddTensor()29 return -new_tensors.size(); in AddTensor()49 gpu_subgraph->new_tensors.clear(); in InitSingleOpSubgraph()
43 std::vector<TensorDescriptor> new_tensors; member
172 new_tensors = torch.nn.modules._functions.BackwardHookFunction.apply(*tensors)173 if len(new_tensors) == 0:176 …grad_fns = [t.grad_fn for t in new_tensors if t.grad_fn is not None and t.grad_fn.name() == "Backw…184 for idx, val in zip(tensors_idx, new_tensors):
309 llvm::SmallVector<mlir::Value, 4> new_tensors; in ConditionalSave() local321 new_tensors.push_back(new_save.getArgument(tensor_index)); in ConditionalSave()355 new_tensors); in ConditionalSave()
339 for (int j = 0; j < gpu_subgraph.new_tensors.size(); ++j) { in ConvertOperations()340 const auto& t = gpu_subgraph.new_tensors[j]; in ConvertOperations()344 std::move(gpu_subgraph.new_tensors[j]); in ConvertOperations()
616 auto new_tensors = broadcast_tensors(std::move(tensors)); in insertExplicitBroadcast() local619 auto new_tensors_it = new_tensors.begin(); in insertExplicitBroadcast()622 AT_ASSERT(new_tensors_it != new_tensors.end()); in insertExplicitBroadcast()
639 new_tensors = [644 "call_function", torch.cat, args=(new_tensors, dim)
597 std::optional<std::vector<at::Tensor>> new_tensors = std::nullopt; in view_func_impl() local599 new_tensors = map_py_func( in view_func_impl()605 if (new_symints.has_value() || new_tensors.has_value()) { in view_func_impl()606 out = (*view_func.clone_and_set(new_symints, new_tensors))(new_base); in view_func_impl()
3012 std::vector<PyObject*> new_tensors; local3029 new_tensors.push_back(jvp);3053 tensorflow::Safe_PyObjectPtr new_tensors_py(PyList_New(new_tensors.size()));3054 for (int i = 0; i < new_tensors.size(); ++i) {3055 PyObject* jvp = new_tensors[i];