Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/profiler/
Dutil.h89 TORCH_API std::string variantShapesToStr(const std::vector<shape>& shapes);
Dutil.cpp220 std::string variantShapesToStr(const std::vector<shape>& shapes) { in variantShapesToStr() function
/external/pytorch/torch/csrc/autograd/
Dprofiler_kineto.cpp77 using torch::profiler::impl::variantShapesToStr;
250 addMetadata("Input Dims", variantShapesToStr(arg_data.shapes)); in operator ()()
251 addMetadata("Input Strides", variantShapesToStr(arg_data.strides)); in operator ()()