Searched refs:onnxShape (Results 1 – 1 of 1) sorted by relevance
337 const onnx::TensorShapeProto onnxShape = info.type().tensor_type().shape(); in ToTensorInfo() local339 for (int i = 0; i < onnxShape.dim_size(); ++i) in ToTensorInfo()341 shapeDims.push_back(CHECKED_NON_NEGATIVE(CHECKED_INT32(onnxShape.dim(i).dim_value()))); in ToTensorInfo()