Searched defs:tensorDimensions (Results 1 – 2 of 2) sorted by relevance
52 std::vector<unsigned int> tensorDimensions(1, 1); in InferOutputShapes() local58 std::vector<unsigned int> tensorDimensions(inputNumDimensions - 1, 0); in InferOutputShapes() local
63 std::vector<unsigned int> tensorDimensions(outputNumDimensions, 0); in InferOutputShapes() local