Home
last modified time | relevance | path

Searched defs:arguments_begin (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.cpp104 const auto arguments_begin = stack->size() - num_arguments; in batchedTensorInplaceForLoopFallback() local
287 const auto arguments_begin = stack->size() - num_arguments; in batchedTensorForLoopFallback() local
428 const auto arguments_begin = stack->size() - num_arguments; in batchedNestedTensorForLoopFallback() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.cpp99 const auto arguments_begin = stack->size() - num_arguments; in batchedTensorInplaceForLoopFallback() local
271 const auto arguments_begin = stack->size() - num_arguments; in batchedTensorForLoopFallback() local
H A DFunctionalizeFallbackKernel.cpp57 const auto arguments_begin = stack->size() - num_arguments; in functionalizeFallback() local
H A DFunctionalTensorWrapper.cpp827 const auto arguments_begin = stack->size() - num_arguments; in functionalize_op_helper() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUFallback.cpp98 const auto arguments_begin = stack->size() - num_arguments; in cpu_fallback() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_eager_fallback.cpp214 const auto arguments_begin = stack->size() - num_arguments; in ts_eager_fallback() local