Searched refs:logical_tensors (Results 1 – 4 of 4) sorted by relevance
190 MultiBatchVmapTransform::logicalToPhysical(ITensorListRef logical_tensors) { in logicalToPhysical() argument193 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()206 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()239 getLevelsAndLargestLogicalDim(TensorList logical_tensors) { in getLevelsAndLargestLogicalDim() argument240 TORCH_INTERNAL_ASSERT(!logical_tensors.empty()); in getLevelsAndLargestLogicalDim()243 for (const auto& tensor : logical_tensors) { in getLevelsAndLargestLogicalDim()256 VmapPhysicalViewVec BroadcastingVmapTransform::logicalToPhysical(TensorList logical_tensors) { in logicalToPhysical() argument258 logical_tensors.size() == 2, in logicalToPhysical()264 auto [levels, largest_logical_dim] = getLevelsAndLargestLogicalDim(logical_tensors); in logicalToPhysical()266 for (const auto& tensor : logical_tensors) { in logicalToPhysical()
60 static VmapPhysicalViewVec logicalToPhysical(ITensorListRef logical_tensors);83 static VmapPhysicalViewVec logicalToPhysical(TensorList logical_tensors);
120 MultiBatchVmapTransform::logicalToPhysical(ITensorListRef logical_tensors) { in logicalToPhysical() argument125 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()141 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()172 VmapPhysicalViewVec BroadcastingVmapTransform::logicalToPhysical(TensorList logical_tensors) { in logicalToPhysical() argument177 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()191 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()196 for (const auto& logical_tensor : logical_tensors) { in logicalToPhysical()
67 static VmapPhysicalViewVec logicalToPhysical(ITensorListRef logical_tensors);90 static VmapPhysicalViewVec logicalToPhysical(TensorList logical_tensors);