Lines Matching defs:RTTBuilder
1452 using RTTBuilder = RankedTensorType::Builder;
1454 RTTBuilder(inputType).dropDim((removeH ? ohIndex : owIndex));
1456 RTTBuilder(kernelType).dropDim((removeH ? khIndex : kwIndex));
1458 RTTBuilder(outputType).dropDim((removeH ? ohIndex : owIndex));
1540 using RTTBuilder = RankedTensorType::Builder;
1542 RTTBuilder(inputType).dropDim((removeH ? 1 : 2));
1544 RTTBuilder(kernelType).dropDim((removeH ? 0 : 1));
1546 RTTBuilder(outputType).dropDim(removeH ? 1 : 2);
1608 using RTTBuilder = RankedTensorType::Builder;
1610 RTTBuilder(inputType).dropDim((removeH ? 0 : 1));
1612 RTTBuilder(kernelType).dropDim((removeH ? 0 : 1));
1614 RTTBuilder(outputType).dropDim(removeH ? 0 : 1);