Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dcollection.cpp277 auto InputOutputEncoder::getInputShapeGenerator() { in getInputShapeGenerator() function in torch::profiler::impl::InputOutputEncoder
457 auto input_shape_getter = inputs_outputs_.getInputShapeGenerator(); in materialize()
H A Dcollection.h480 auto getInputShapeGenerator();