Searched refs:nativeShape (Results 1 – 3 of 3) sorted by relevance
117 NativeShapeFnType nativeShape = nullptr; member119 nativeShape = std::move(fn); in setNativeShapeFn()126 nativeShape = [=](Operation *) -> Optional<SmallVector<int64_t, 4>> { in setNativeShape()
317 SmallVector<int64_t, 4> nativeShape( in runOnOperation() local320 nativeShape[nativeShape.size() - 1] = lhsType.isF16() ? 4 : 2; in runOnOperation()321 return nativeShape; in runOnOperation()
146 assert(options.nativeShape && in getTargetShape()155 Optional<SmallVector<int64_t, 4>> targetShape = options.nativeShape(op); in getTargetShape()