Searched refs:getInputShapeGenerator (Results 1 – 2 of 2) sorted by relevance
277 auto InputOutputEncoder::getInputShapeGenerator() { in getInputShapeGenerator() function in torch::profiler::impl::InputOutputEncoder457 auto input_shape_getter = inputs_outputs_.getInputShapeGenerator(); in materialize()
480 auto getInputShapeGenerator();