Home
last modified time | relevance | path

Searched defs:physical_tensors (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyBatchingRegistrations.cpp587 auto physical_tensors = fmap( in block_diag_batching_rule() local
615 auto physical_tensors = fmap( in stack_batching_rule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyVmapTransforms.cpp204 std::vector<Tensor> physical_tensors; in logicalToPhysical() local
H A DLegacyBatchingRegistrations.cpp933 auto physical_tensors = fmap( in cat_batching_rule() local
943 auto physical_tensors = fmap( in stack_batching_rule() local