Home
last modified time | relevance | path

Searched refs:cache_shape_function (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_cache.h46 TORCH_API void cache_shape_function(
H A Dsymbolic_shape_cache.cpp100 TORCH_API void cache_shape_function( in cache_shape_function() function
H A Dsymbolic_shape_analysis.cpp1170 cache_shape_function(schema, inputs, merged_res); in calculateSymbolicShapesOnOp()
1179 cache_shape_function(schema, inputs, res.value()); in calculateSymbolicShapesOnOp()