Home
last modified time | relevance | path

Searched refs:warnFallback (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/
DLegacyBatchedFallback.cpp69 static void warnFallback(const c10::FunctionSchema& schema) { in warnFallback() function
95 warnFallback(schema); in batchedTensorInplaceForLoopFallback()
267 warnFallback(schema); in batchedTensorForLoopFallback()
/external/pytorch/aten/src/ATen/functorch/
DBatchedFallback.cpp75 static void warnFallback(const c10::FunctionSchema& schema, bool is_inplace, bool is_nested=false) { in warnFallback() function
100 warnFallback(schema, /*in_place*/true); in batchedTensorInplaceForLoopFallback()
285 warnFallback(schema, /*in_place*/false); in batchedTensorForLoopFallback()
426 warnFallback(schema, /*in_place*/false, /*is_nested*/true); in batchedNestedTensorForLoopFallback()