Home
last modified time | relevance | path

Searched defs:stack_inputs (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp2076 auto stack_inputs = stack_ivals.slice(nOutputs_); in runWithAllocatedOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py801 def stack_inputs(self, stack_indices=None, tile_variants=False): member in _PforInput