Searched refs:tileSizeComputationFunction (Results 1 – 5 of 5) sorted by relevance
36 SCFTileSizeComputationFunction tileSizeComputationFunction = nullptr; member40 tileSizeComputationFunction = std::move(fun); in setTileSizeComputationFunction()47 tileSizeComputationFunction = [=](OpBuilder &, Operation *) { return ts; }; in setTileSizes()
31 assert(!tileSizeComputationFunction && "tile sizes already set"); in setTileSizes()33 tileSizeComputationFunction = [tileSizes](OpBuilder &b, Operation *op) { in setTileSizes()166 if (!options.tileSizeComputationFunction) { in returningMatchAndRewrite()184 options.tileSizeComputationFunction(rewriter, op); in returningMatchAndRewrite()
123 assert(!tileSizeComputationFunction && "tile sizes already set"); in setTileSizes()125 tileSizeComputationFunction = [tileSizes](OpBuilder &b, Operation *op) { in setTileSizes()138 assert(!tileSizeComputationFunction && "tile sizes already set"); in scalarizeDynamicDims()139 tileSizeComputationFunction = [](OpBuilder &b, Operation *op) { in scalarizeDynamicDims()
515 if (!options.tileSizeComputationFunction) in tileLinalgOpImpl()523 options.tileSizeComputationFunction(b, op); in tileLinalgOpImpl()560 options.tileSizeComputationFunction(builder, op); in tilePadOp()
594 TileSizeComputationFunction tileSizeComputationFunction = nullptr; member598 tileSizeComputationFunction = std::move(fun); in setTileSizeComputationFunction()605 tileSizeComputationFunction = [=](OpBuilder &, Operation *) { return ts; }; in setTileSizes()