| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 106 DenseIntElementsAttr Builder::getBoolVectorAttr(ArrayRef<bool> values) { in getBoolVectorAttr()
107 return DenseIntElementsAttr::get( in getBoolVectorAttr()
112 DenseIntElementsAttr Builder::getI32VectorAttr(ArrayRef<int32_t> values) { in getI32VectorAttr()
113 return DenseIntElementsAttr::get( in getI32VectorAttr()
118 DenseIntElementsAttr Builder::getI64VectorAttr(ArrayRef<int64_t> values) { in getI64VectorAttr()
119 return DenseIntElementsAttr::get( in getI64VectorAttr()
124 DenseIntElementsAttr Builder::getI32TensorAttr(ArrayRef<int32_t> values) { in getI32TensorAttr()
125 return DenseIntElementsAttr::get( in getI32TensorAttr()
131 DenseIntElementsAttr Builder::getI64TensorAttr(ArrayRef<int64_t> values) { in getI64TensorAttr()
132 return DenseIntElementsAttr::get( in getI64TensorAttr()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Shape/IR/ |
| D | Shape.cpp | 110 value.cast<DenseIntElementsAttr>()); in materializeConstant()
318 operands[1].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
326 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
347 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
349 operands[1].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
441 operands[0].cast<DenseIntElementsAttr>().getNumElements() == 0) in fold()
444 operands[1].cast<DenseIntElementsAttr>().getNumElements() == 0) in fold()
449 operands[0].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
451 operands[1].cast<DenseIntElementsAttr>().getValues<int64_t>()); in fold()
532 auto lhs = operands[0].dyn_cast_or_null<DenseIntElementsAttr>(); in fold()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_exporter.h | 36 mlir::DenseIntElementsAttr input);
41 llvm::Optional<mlir::DenseIntElementsAttr> optional_attr);
|
| D | attribute_exporter.cc | 91 mlir::DenseIntElementsAttr input) { in ConvertReplicaGroups()
118 llvm::Optional<mlir::DenseIntElementsAttr> optional_attr) { in ConvertNx2Attribute()
120 mlir::DenseIntElementsAttr attr = *optional_attr; in ConvertNx2Attribute()
|
| D | hlo_function_importer.h | 141 mlir::DenseIntElementsAttr ConvertDimensions(
145 mlir::DenseIntElementsAttr Convert(llvm::ArrayRef<int64_t> elements);
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 41 class DenseIntElementsAttr; variable
124 DenseIntElementsAttr getBoolVectorAttr(ArrayRef<bool> values);
125 DenseIntElementsAttr getI32VectorAttr(ArrayRef<int32_t> values);
126 DenseIntElementsAttr getI64VectorAttr(ArrayRef<int64_t> values);
131 DenseIntElementsAttr getI32TensorAttr(ArrayRef<int32_t> values);
132 DenseIntElementsAttr getI64TensorAttr(ArrayRef<int64_t> values);
133 DenseIntElementsAttr getIndexTensorAttr(ArrayRef<int64_t> values);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | hlo_utils.cc | 25 DenseIntElementsAttr getBroadcastDimensionsAttr(Builder *b, Value x, Value y, in getBroadcastDimensionsAttr()
54 return DenseIntElementsAttr::get(type, broadcastDimensions); in getBroadcastDimensionsAttr()
152 bool IsSequenceStartingWith0(DenseIntElementsAttr attr) { in IsSequenceStartingWith0()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 133 DenseIntElementsAttr depth_attr; in Verify()
160 DenseIntElementsAttr depth_attr; in InferOneHotOpType()
257 auto value_attr = const_op.value().dyn_cast<DenseIntElementsAttr>(); in fold()
332 auto values = mlir::DenseIntElementsAttr::get(type, shuffled_paddings); in FoldOperandsPermutation()
498 DenseIntElementsAttr start_val; in build()
499 DenseIntElementsAttr limit_val; in build()
500 DenseIntElementsAttr delta_val; in build()
542 return DenseIntElementsAttr::get(output_type, rank); in fold()
590 DenseIntElementsAttr shape_attr; in GetReshapeOutputType()
1098 DenseIntElementsAttr begin_indices; in Verify()
[all …]
|
| D | tf_ops_a_m.cc | 314 DenseIntElementsAttr crops_attr; in Verify()
591 DenseIntElementsAttr &s0, DenseIntElementsAttr &s1, in ExtractInputConstShape()
656 DenseIntElementsAttr s0, s1; in Verify()
687 DenseIntElementsAttr s0, s1; in fold()
713 return DenseIntElementsAttr::get(type, i32_shape); in fold()
716 return DenseIntElementsAttr::get(type, shape); in fold()
741 DenseIntElementsAttr branch; in matchAndRewrite()
1086 DenseIntElementsAttr axis_attr; in matchAndRewrite()
1178 loc, DenseIntElementsAttr::get(axis_type, hoist_params->lhs_axis)); in matchAndRewrite()
1180 loc, DenseIntElementsAttr::get(axis_type, hoist_params->rhs_axis)); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 137 static DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr()
141 return DenseIntElementsAttr::get(ty, values); in GetI64ElementsAttr()
145 static DenseIntElementsAttr GetI64ElementsAttr(ArrayAttr attr) { in GetI64ElementsAttr()
149 return DenseIntElementsAttr::get(ty, attr.getValue()); in GetI64ElementsAttr()
153 static DenseIntElementsAttr GetI32ElementsAttr(ArrayRef<int32_t> values, in GetI32ElementsAttr()
157 return DenseIntElementsAttr::get(ty, values); in GetI32ElementsAttr()
162 static DenseIntElementsAttr GetI64ElementsAttrForSeq(int start, int end, in GetI64ElementsAttrForSeq()
171 return DenseIntElementsAttr::get(ty, vals); in GetI64ElementsAttrForSeq()
176 static DenseIntElementsAttr GetI64ElementsAttrForValue(int size, int64_t val, in GetI64ElementsAttrForValue()
179 return DenseIntElementsAttr::get(ty, val); in GetI64ElementsAttrForValue()
[all …]
|
| D | mhlo_to_lhlo_with_xla.h | 162 DenseIntElementsAttr GetI64DenseElementsAttr(const T& container) { in GetI64DenseElementsAttr()
167 DenseIntElementsAttr GetWindowElements( in GetWindowElements()
178 static mlir::DenseIntElementsAttr GetLayoutAttribute(
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 69 DenseIntElementsAttr broadcast_dimensions_attr) { in GetBroadcastType()
132 DenseIntElementsAttr broadcast_dimensions = in InferBroadcastBinaryOpReturnTypeComponents()
134 .dyn_cast_or_null<DenseIntElementsAttr>(); in InferBroadcastBinaryOpReturnTypeComponents()
167 .dyn_cast_or_null<DenseIntElementsAttr>(); in ReifyBroadcastBinaryOpReturnTypeShapes()
219 DenseIntElementsAttr broadcast_dimensions, in build()
304 Value right, DenseIntElementsAttr broadcast_dimensions) { \
|
| D | hlo_ops.cc | 111 DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr()
115 return DenseIntElementsAttr::get(ty, values); in GetI64ElementsAttr()
120 DenseIntElementsAttr BuildSliceLimits(DenseIntElementsAttr start_indices, in BuildSliceLimits()
121 DenseIntElementsAttr slice_sizes, in BuildSliceLimits()
203 DenseIntElementsAttr index; in matchAndRewrite()
280 return DenseIntElementsAttr::get<int32_t>( in fold()
322 auto broadcast_attr = DenseIntElementsAttr::get( in matchAndRewrite()
410 auto broadcast_attr = DenseIntElementsAttr::get( in matchAndRewrite()
1437 DenseIntElementsAttr slice_limits = BuildSliceLimits( in matchAndRewrite()
1439 DenseIntElementsAttr slice_strides = in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/ |
| D | hlo_utils.h | 32 mlir::DenseIntElementsAttr getBroadcastDimensionsAttr(mlir::Builder* b,
92 bool IsSequenceStartingWith0(DenseIntElementsAttr attr);
|
| D | broadcast_utils.h | 37 DenseIntElementsAttr broadcast_dims);
|
| /external/llvm-project/mlir/include/mlir/Bindings/Python/ |
| D | Attributes.td | 31 def : PythonAttr<"::mlir::DenseIntElementsAttr", "_ir.DenseIntElementsAttr">;
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lower_general_dot.cc | 33 using mlir::DenseIntElementsAttr;
76 DenseIntElementsAttr::get(transpose_permutation_type, in TransposeReshape()
78 .cast<DenseIntElementsAttr>(); in TransposeReshape()
|
| D | optimize_mhlo.cc | 44 static DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, in GetI64ElementsAttr()
48 return DenseIntElementsAttr::get(ty, values); in GetI64ElementsAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 229 DenseIntElementsAttr input_spatial_dimensions = in IsSupportedConvOp()
241 DenseIntElementsAttr output_spatial_dimensions = in IsSupportedConvOp()
255 DenseIntElementsAttr kernal_spatial_dimensions = in IsSupportedConvOp()
273 DenseIntElementsAttr strides = slice_op.strides(); in matchAndRewrite()
294 slice_op.getLoc(), DenseIntElementsAttr::get(ty, size_values)); in matchAndRewrite()
395 DotDimensionsInfo(ShapedType type, DenseIntElementsAttr batch_dimensions, in DotDimensionsInfo()
396 DenseIntElementsAttr contracting_dimensions) { in DotDimensionsInfo()
479 DenseIntElementsAttr::get( in ConvertDot()
497 DenseIntElementsAttr::get( in ConvertDot()
626 DenseIntElementsAttr dimension = reduce_op.dimensions(); in matchAndRewrite()
[all …]
|
| D | tpu_space_to_depth_pass.cc | 173 auto paddings = DenseIntElementsAttr::get(padding_type, values); in HandlePad()
225 auto paddings = DenseIntElementsAttr::get(padding_type, values); in GetPadOpForConv2DFilter()
245 auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); in GetReshapeOpForConv2DFilter()
256 auto permute_attr = DenseIntElementsAttr::get(permute_type, permutation); in GetTransposeOpForConv2DFilter()
321 DenseIntElementsAttr::get( in GetSliceOpForConv2DBackPropFilter()
329 DenseIntElementsAttr::get(start_position_type, slice_start_position)); in GetSliceOpForConv2DBackPropFilter()
346 DenseIntElementsAttr::get( in HandleConv2DBackPropFilter()
|
| /external/llvm-project/flang/lib/Optimizer/Dialect/ |
| D | FIROps.cpp | 993 mlir::DenseIntElementsAttr ranges, in getSubOperands()
1009 pos, operands, targetOffsetAttr.second.cast<DenseIntElementsAttr>(), in getMutableSuccessorOperands()
1013 static unsigned denseElementsSize(mlir::DenseIntElementsAttr attr) { in denseElementsSize()
1035 auto a = getAttrOfType<mlir::DenseIntElementsAttr>(getTargetOffsetAttr()); in getSuccessorOperands()
1037 getAttrOfType<mlir::DenseIntElementsAttr>(getOperandSegmentSizeAttr()); in getSuccessorOperands()
1043 getAttrOfType<mlir::DenseIntElementsAttr>(getTargetOffsetAttr())); in targetOffsetSize()
1052 auto a = getAttrOfType<mlir::DenseIntElementsAttr>(getCompareOffsetAttr()); in getCompareOperands()
1059 auto a = getAttrOfType<mlir::DenseIntElementsAttr>(getCompareOffsetAttr()); in getCompareOperands()
1061 getAttrOfType<mlir::DenseIntElementsAttr>(getOperandSegmentSizeAttr()); in getCompareOperands()
1074 auto a = getAttrOfType<mlir::DenseIntElementsAttr>(getTargetOffsetAttr()); in getSuccessorOperands()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_utils.td | 25 def CastIntElementsAttr : NativeCodeCall<"$0.cast<DenseIntElementsAttr>()">;
42 def NullDenseIntElementsAttr : NativeCodeCall<"DenseIntElementsAttr()">;
|
| /external/llvm-project/mlir/test/Bindings/Python/ |
| D | ir_attributes.py | 207 a = DenseIntElementsAttr(raw)
222 a = DenseIntElementsAttr(raw)
|
| /external/llvm-project/mlir/lib/Bindings/Python/mlir/dialects/ |
| D | __init__.py | 17 segments = _cext.ir.DenseIntElementsAttr(raw_segments)
|
| /external/llvm-project/mlir/unittests/TableGen/ |
| D | StructsGenTest.cpp | 36 mlir::DenseIntElementsAttr::get(elementsType, {1, 2, 3, 4, 5, 6}); in getTestStruct()
111 mlir::DenseIntElementsAttr::get(elementsType, ArrayRef<int64_t>{1, 2, 3}); in TEST()
|