Searched refs:insertShapeProfile (Results 1 – 2 of 2) sorted by relevance
134 void ProfilingRecord::insertShapeProfile( in insertShapeProfile() function in torch::jit::ProfilingRecord274 insertShapeProfile(n, offset, i->type()); in instrumentBlock()294 insertShapeProfile(block->return_node(), offset, i->type()); in instrumentBlock()
201 void insertShapeProfile(Node* n, size_t offset, const TypePtr& input_type);