Home
last modified time | relevance | path

Searched refs:kDynamicSize (Results 1 – 25 of 35) sorted by relevance

12

/external/llvm-project/mlir/test/EDSC/
Dbuilder-api-test.cpp316 MemRefType::get({ShapedType::kDynamicSize, ShapedType::kDynamicSize, in TEST_FUNC()
317 ShapedType::kDynamicSize}, in TEST_FUNC()
541 {ShapedType::kDynamicSize, ShapedType::kDynamicSize}, i32Type, {}, 0); in TEST_FUNC()
616 {ShapedType::kDynamicSize, ShapedType::kDynamicSize}, f32Type, {}, 0); in TEST_FUNC()
723 MemRefType::get({ShapedType::kDynamicSize, ShapedType::kDynamicSize, in TEST_FUNC()
724 ShapedType::kDynamicSize}, in TEST_FUNC()
797 auto memrefType = MemRefType::get({ShapedType::kDynamicSize}, in TEST_FUNC()
868 {ShapedType::kDynamicSize, ShapedType::kDynamicSize}, f32Type, {}, 0); in TEST_FUNC()
921 {ShapedType::kDynamicSize, ShapedType::kDynamicSize}, f32Type, {}, 0); in TEST_FUNC()
958 {ShapedType::kDynamicSize, ShapedType::kDynamicSize}, f32Type, {}, 0); in TEST_FUNC()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dverification_utils.cc27 if (dim != ShapedType::kDynamicSize) continue; in VerifyShapeOfReshapeOp()
Dxla_sharding_util.cc70 mlir::ShapedType::kDynamicSize) { in CreateSplitOp()
124 mlir::ShapedType::kDynamicSize) { in CreateConcatOp()
471 if (output_shape[dimension_index] == mlir::ShapedType::kDynamicSize) { in ValidateAndGetTiledExecuteOutputShape()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dtransform_unranked_hlo.cc117 RankedTensorType::get({ShapedType::kDynamicSize}, operandElementTy); in matchAndRewrite()
127 RankedTensorType::get({ShapedType::kDynamicSize}, resultElementTy); in matchAndRewrite()
341 {RankedTensorType::kDynamicSize}, builder.getIndexType()); in extendToBroadcastShape()
355 {RankedTensorType::kDynamicSize}, builder.getIndexType()); in createBroadcastToKnownRank()
381 targeted_rank, RankedTensorType::kDynamicSize); in createRankSpecializedBroadcastAndOp()
415 auto extent_tensor_type = RankedTensorType::get({ShapedType::kDynamicSize}, in HandleBroadcastAndOp()
Dhlo_legalize_to_lhlo.cc63 if (shape_element.value() != ShapedType::kDynamicSize) continue; in InsertDynamicAllocAndDealloc()
352 MemRefType::kDynamicSize); in InsertDynamicMemrefCastOp()
Dlegalize_to_linalg.cc100 if (en.value() != ShapedType::kDynamicSize) continue; in ExtractDynamicSizes()
1084 return a == ShapedType::kDynamicSize || b == ShapedType::kDynamicSize || in GetDotOperationType()
Dlhlo_legalize_to_parallel_loops.cc80 return dim == ShapedType::kDynamicSize in GetStaticOrDynamicDim()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc159 int64_t depth_val = ShapedType::kDynamicSize; in InferOneHotOpType()
596 ShapedType::kDynamicSize); in GetReshapeOutputType()
611 if (size == ShapedType::kDynamicSize) { in GetReshapeOutputType()
1320 SmallVector<int64_t, 4> return_shape(input_rank, ShapedType::kDynamicSize); in inferReturnTypes()
1323 if (block_rank == ShapedType::kDynamicSize) { in inferReturnTypes()
1343 if (input_shape[i + 1] == ShapedType::kDynamicSize) { in inferReturnTypes()
1344 return_batch = ShapedType::kDynamicSize; in inferReturnTypes()
1346 if (return_batch == ShapedType::kDynamicSize) { in inferReturnTypes()
1347 return_shape[1 + i] = ShapedType::kDynamicSize; in inferReturnTypes()
1441 if (input_dim_size == ShapedType::kDynamicSize) return success(); in Verify()
[all …]
Dtf_ops_a_m.cc225 if (output_shape[i] != ShapedType::kDynamicSize && in Verify()
245 if (expected_out_row_dim != ShapedType::kDynamicSize && in Verify()
246 out_row_dim != ShapedType::kDynamicSize && in Verify()
251 if (expected_out_col_dim != ShapedType::kDynamicSize && in Verify()
252 out_col_dim != ShapedType::kDynamicSize && in Verify()
279 llvm::SmallVector<int64_t, 4> input_shape(4, ShapedType::kDynamicSize); in Verify()
287 if (input_batch != ShapedType::kDynamicSize && in Verify()
341 return dim_a != ShapedType::kDynamicSize && in Verify()
342 dim_b != ShapedType::kDynamicSize; in Verify()
1633 SmallVector<int64_t, 4> return_shape(num_dims, ShapedType::kDynamicSize); in inferConvReturnTypes()
[all …]
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DBufferize.cpp34 if (dim.value() == TensorType::kDynamicSize) { in cloneMemref()
88 if (shapeElement.value() != ShapedType::kDynamicSize) in allocateBuffersForResults()
DSparsification.cpp473 auto dynShape = {ShapedType::kDynamicSize}; in genBuffers()
483 if (shape[d] == TensorType::kDynamicSize) { in genBuffers()
504 auto sparseTp = MemRefType::get({ShapedType::kDynamicSize}, in genBuffers()
DPromotion.cpp123 ShapedType::kDynamicSize); in defaultAllocBufferCallBack()
/external/llvm-project/mlir/lib/Conversion/ShapeToStandard/
DShapeToStandard.cpp103 RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
187 Type resultTy = RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
246 RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/IR/
DBuiltinTypes.h334 static constexpr int64_t kDynamicSize = -1; variable
395 return dSize == kDynamicSize; in isDynamic()
/external/llvm-project/mlir/lib/Interfaces/
DViewLikeInterface.cpp173 ShapedType::kDynamicSize, sizesInfo) || in parseOffsetsSizesAndStrides()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc147 llvm::ArrayRef<Type>{RankedTensorType::get({ShapedType::kDynamicSize}, in BuildGetLayout()
Dshape_inference.cc991 current_dim != ShapedType::kDynamicSize) in CanWhileTypeBeRefinedWith()
1281 dims.push_back(ShapedType::kDynamicSize); in GetCompatibleRankedTensorType()
/external/llvm-project/mlir/lib/Dialect/StandardOps/IR/
DOps.cpp1405 RankedTensorType::kDynamicSize); in fold()
1410 if (dim == RankedTensorType::kDynamicSize) in fold()
1822 if (dim != RankedTensorType::kDynamicSize) { in matchAndRewrite()
1828 newShape.push_back(RankedTensorType::kDynamicSize); in matchAndRewrite()
2340 SmallVector<int64_t, 4> staticSizesVector(rank, ShapedType::kDynamicSize); in build()
2486 if (shapeSize == ShapedType::kDynamicSize) in verify()
3138 staticSizesVector.assign(rank, ShapedType::kDynamicSize); in build()
3170 staticSizesVector.assign(rank, ShapedType::kDynamicSize); in build()
3766 SmallVector<int64_t, 4> staticSizesVector(rank, ShapedType::kDynamicSize); in build()
3861 SmallVector<int64_t, 4> staticSizesVector(rank, ShapedType::kDynamicSize); in build()
/external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/
DExpandOps.cpp102 SmallVector<int64_t, 2> staticSizes(rank, ShapedType::kDynamicSize); in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/Dialect/Shape/IR/
DShapeBase.td111 BuildableType<"::mlir::RankedTensorType::get({ShapedType::kDynamicSize}, "
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc599 constexpr int64_t kDynamicSize = -1; in VerifyConcatenationOpTypes() local
614 result_dim_sizes[axis] = kDynamicSize; in VerifyConcatenationOpTypes()
631 result_dim_sizes[axis] = kDynamicSize; in VerifyConcatenationOpTypes()
1368 ShapedType::kDynamicSize); in GetReshapeOutputType()
1383 if (size == ShapedType::kDynamicSize) { in GetReshapeOutputType()
/external/llvm-project/mlir/include/mlir/Interfaces/
DViewLikeInterface.td51 `ShapedType::kDynamicStrideOrOffset` (resp. `ShapedType::kDynamicSize`,
/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp750 size = ShapedType::kDynamicSize; in computeReshapeCollapsedType()
943 if (llvm::is_contained(band, ShapedType::kDynamicSize)) in computeTensorReshapeCollapsedType()
944 size = ShapedType::kDynamicSize; in computeTensorReshapeCollapsedType()
/external/llvm-project/mlir/lib/Dialect/SPIRV/
DSPIRVLowering.cpp200 if (llvm::is_contained(dims, ShapedType::kDynamicSize) || in getTypeNumBytes()
/external/llvm-project/mlir/lib/IR/
DBuiltinTypes.cpp193 constexpr int64_t ShapedType::kDynamicSize; member in ShapedType

12