Searched defs:stretchDim (Results 1 – 4 of 4) sorted by relevance
93 const auto stretchDim = std::find(targetShape.begin(), targetShape.end(), -1); in CreateOutputTensorShape() local
423 const auto stretchDim = std::find(targetDims.begin(), targetDims.end(), -1); in ComputeReshapeInfo() local
2584 const auto stretchDim = std::find(targetDimsIn.begin(), targetDimsIn.end(), -1); in OutputShapeOfReshape() local
2949 const auto stretchDim = std::find(targetDimsIn.begin(), targetDimsIn.end(), -1); in OutputShapeOfReshape() local