Searched refs:targetDimsSize (Results 1 – 2 of 2) sorted by relevance
383 bool reshapePrepare(const Shape& input, const int32_t* targetDims, const int32_t targetDimsSize, in reshapePrepare() argument391 std::vector<uint32_t> outDims(targetDimsSize); in reshapePrepare()394 for (int32_t i = 0; i < targetDimsSize; ++i) { in reshapePrepare()
290 bool reshapePrepare(const Shape& input, const int32_t* targetDims, const int32_t targetDimsSize,