Home
last modified time | relevance | path

Searched refs:getNumDynamicDims (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ir/
Dtf_framework_ops.cc71 unsigned result_dyn_dims = op.getType().getNumDynamicDims(); in Verify()
/external/llvm-project/mlir/lib/Transforms/Utils/
DUtils.cpp455 if (memrefType.getNumDynamicDims() > 0) in normalizeMemRefType()
/external/llvm-project/mlir/include/mlir/IR/
DBuiltinTypes.h368 int64_t getNumDynamicDims() const;
/external/llvm-project/mlir/lib/IR/
DBuiltinTypes.cpp267 int64_t ShapedType::getNumDynamicDims() const { in getNumDynamicDims() function in ShapedType
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc177 if (!ranked_output.hasRank() || ranked_output.getNumDynamicDims() != 0) { in matchAndRewrite()
Doptimize_patterns.td490 "$0.getType().cast<RankedTensorType>().getNumDynamicDims() <= 1">>;
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp272 memRefType.getNumDynamicDims()) in verifyAllocLikeOp()
344 newMemRefType.getNumDynamicDims()); in matchAndRewrite()
1761 if (op.getNumOperands() != resultTy.getNumDynamicDims()) in verify()
4223 unsigned numDynamicDims = viewType.getNumDynamicDims(); in verify()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc289 if (!arg_ty || !arg_ty.hasRank() || arg_ty.getNumDynamicDims() != 1 || in fold()