Searched refs:new_tensors (Results 1 – 5 of 5) 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
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()
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];