Home
last modified time | relevance | path

Searched defs:new_tensors (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Dsubgraph.h43 std::vector<TensorDescriptor> new_tensors; member
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsave_restore_spmd_expander.cc309 llvm::SmallVector<mlir::Value, 4> new_tensors; in ConditionalSave() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_fuser.cpp616 auto new_tensors = broadcast_tensors(std::move(tensors)); in insertExplicitBroadcast() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp597 std::optional<std::vector<at::Tensor>> new_tensors = std::nullopt; in view_func_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc3012 std::vector<PyObject*> new_tensors; local