Searched refs:outShape (Results 1 – 1 of 1) sorted by relevance
355 SmallVector<int64_t> outShape; in inferReturnTypeComponents() local433 SmallVector<int64_t> outShape; in inferReturnTypeComponents() local445 outShape[1] = outShape[1] == ShapedType::kDynamicSize in inferReturnTypeComponents()447 : outShape[1]; in inferReturnTypeComponents()464 SmallVector<int64_t> outShape; in inferReturnTypeComponents() local468 outShape[0] = lhsShape.getDimSize(0); in inferReturnTypeComponents()469 outShape[1] = lhsShape.getDimSize(1); in inferReturnTypeComponents()473 outShape[0] = outShape[0] == ShapedType::kDynamicSize in inferReturnTypeComponents()475 : outShape[0]; in inferReturnTypeComponents()884 outShape.resize(outRank, 1); in REDUCE_SHAPE_INFER()[all …]