Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyVmapTransforms.cpp190 MultiBatchVmapTransform::logicalToPhysical(ITensorListRef logical_tensors) { in logicalToPhysical()
239 getLevelsAndLargestLogicalDim(TensorList logical_tensors) { in getLevelsAndLargestLogicalDim()
256 VmapPhysicalViewVec BroadcastingVmapTransform::logicalToPhysical(TensorList logical_tensors) { in logicalToPhysical()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyVmapTransforms.cpp120 MultiBatchVmapTransform::logicalToPhysical(ITensorListRef logical_tensors) { in logicalToPhysical()
172 VmapPhysicalViewVec BroadcastingVmapTransform::logicalToPhysical(TensorList logical_tensors) { in logicalToPhysical()