Searched defs:dimensionSizes (Results 1 – 2 of 2) sorted by relevance
40 TensorShape::TensorShape(const unsigned int numDimensions, const unsigned int* const dimensionSizes) in TensorShape()60 const unsigned int* const dimensionSizes, in TensorShape()360 const unsigned int* dimensionSizes, in TensorInfo()385 const unsigned int* dimensionSizes, in TensorInfo()
45 std::vector<unsigned int> dimensionSizes(inputNumDimensions + 1, 0); in InferOutputShapes() local