Lines Matching defs:dimensions

1615     ValueRange inits, ArrayRef<int64_t> dimensions,
1618   build(builder, result, TypeRange{}, inputs, inits, dimensions);
1693             return parseDenseI64ArrayAttr(parser, attributes, "dimensions");
1772              << "dimensions for reduction should be in the range [0, "
1781   // Input dimensions that will be left after the reduction.
1789     return emitOpError() << "number of dimensions after reduction "
1796     return emitOpError() << "init dimensions [" << initDims
1797                          << "] doesn't match input dimensions after reduction ["
2012     ArrayRef<int64_t> dimensions = broadcastOp.getDimensions();
2015     // Get new perms and new dimensions.
2016     SmallVector<int64_t> resultPerms = dropDims(perms, dimensions);
2019     unsigned dimensionSize = dimensions.size();
2021       resultDimensions.push_back(invertPerm[dimensions[i]]);
2069                         DenseI64ArrayAttr dimensions,
2073   result.addAttribute(getDimensionsAttrName(result.name), dimensions);
2087                         ArrayRef<int64_t> dimensions,
2089   build(builder, result, input, init, builder.getDenseI64ArrayAttr(dimensions),
2096             return parseDenseI64ArrayAttr(parser, attributes, "dimensions");
2132     return emitOpError() << "input rank plus added dimensions does not "
2135                          << ", dimensions size: " << dimensionsRef.size()
2153     // This dimensions is mapped from the input. Init and input dims should
2447 /// dimensions to their affine dim expressions.
2753 // -- identity with inputRank dimensions.
2761 // -- identity with inputRank dimensions.