Home
last modified time | relevance | path

Searched refs:tensors_to_compute (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dloopnest.h29 const std::vector<Tensor>& tensors_to_compute);
587 const std::vector<Tensor>& tensors_to_compute);
H A Dloopnest.cpp50 const std::vector<Tensor>& tensors_to_compute) { in LoopNest() argument
51 initialize(output_tensors, tensors_to_compute); in LoopNest()
624 const std::vector<Tensor>& tensors_to_compute) { in initialize() argument
630 for (const Tensor& t : tensors_to_compute) { in initialize()