| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 219 for (auto it : llvm::zip(dnums.start_index_map().getIntValues(), in matchAndRewrite()
220 index.getIntValues())) { in matchAndRewrite()
242 dnums.collapsed_slice_dims().getIntValues(), in matchAndRewrite()
1527 for (auto it : llvm::zip(rhsVal.getIntValues(), lhsVal.getIntValues())) { in fold()
1571 for (auto it : llvm::zip(rhsVal.getIntValues(), lhsVal.getIntValues())) { in fold()
1604 for (auto it : llvm::zip(rhsVal.getIntValues(), lhsVal.getIntValues())) { in fold()
1977 if (llvm::all_of(edge_padding_low().getIntValues(), is_zero) && in fold()
1978 llvm::all_of(edge_padding_high().getIntValues(), is_zero) && in fold()
1979 llvm::all_of(interior_padding().getIntValues(), is_zero)) in fold()
1984 if (llvm::all_of(edge_padding_low().getIntValues(), is_negative) && in fold()
[all …]
|
| D | chlo_ops.cc | 99 for (const APInt& int_value : broadcast_dimensions_attr.getIntValues()) { in GetBroadcastType()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | broadcast_utils.cc | 48 broadcast_dims.getIntValues().begin()); in IsLegalNumpyRankedBroadcast()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_gather_to_torch_index_select.cc | 88 for (auto it : llvm::enumerate(gather.slice_sizes().getIntValues())) { in matchAndRewrite()
|
| D | optimize_mhlo.cc | 164 for (auto shapeValue : gather.slice_sizes().getIntValues()) { in matchAndRewrite()
|
| D | lhlo_legalize_to_parallel_loops.cc | 231 for (const auto& rdim : reduce_op.dimensions().getIntValues()) { in CreateReduceOpInNestedParallelLoops()
557 s_and_s_op.window_dimensions()->getIntValues()) { in InsertWindowLoops()
|
| D | legalize_to_linalg.cc | 468 enumerate(broadcast_op.broadcast_dimensions().getIntValues())) { in getIndexingMaps()
610 enumerate(op.broadcast_dimensions().getIntValues())) { in InsertReshapeIfNecessary()
941 for (const auto& dim : dimensions_attr.getIntValues()) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | layout_optimization.cc | 234 for (const auto& value : perm_attr.getIntValues()) in MoveTransposeBefore()
366 for (const auto& value : attr.getIntValues()) in MoveTransposeAfter()
|
| D | lower_tf.cc | 1063 for (auto val : block_shape.getIntValues()) { in matchAndRewrite()
1131 auto block_shape_values = llvm::to_vector<4>(block_shape.getIntValues()); in matchAndRewrite()
1159 auto crop_values = llvm::to_vector<4>(crops.getIntValues()); in matchAndRewrite()
1325 for (auto val : out_size_cst.getIntValues()) { in matchAndRewrite()
|
| D | legalize_hlo.cc | 1001 for (auto s : attr.getIntValues()) { in IsIotaAttr()
|
| D | shape_inference.cc | 860 int64_t val = (*dea.getIntValues().begin()).getSExtValue(); in ComputeOutputAsShape()
|
| /external/llvm-project/mlir/unittests/TableGen/ |
| D | StructsGenTest.cpp | 161 for (const auto &valIndexIt : llvm::enumerate(denseAttr.getIntValues())) { in TEST()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/ |
| D | LinalgToSPIRV.cpp | 130 if (llvm::any_of(llvm::drop_begin(localSize.getIntValues(), 1), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 128 (*const_op.value().cast<DenseElementsAttr>().getIntValues().begin()) in matchAndRewrite()
|
| D | optimize.cc | 277 for (auto a : axes_attr.getIntValues()) { in ShapeMatchesReduceWithKeepAxes()
903 for (const auto &dim : perm_values_attr.getIntValues()) in matchAndRewrite()
|
| D | lower_static_tensor_list.cc | 422 auto int_values = dense_elem_attr.getIntValues(); in matchAndRewrite()
774 for (const auto &dim : dense_elem_attr.getIntValues()) in matchAndRewrite()
|
| D | prepare_tf.cc | 730 for (const auto &idx : dense_elem_attr.getIntValues()) { in PadStridedSliceAttributeArray()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | shape_inference_utils.cc | 394 for (auto element : llvm::enumerate(elements.getIntValues())) in InferReturnTypeComponentsForTFOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 322 auto crops_range = crops_attr.getIntValues(); in Verify()
597 for (auto s : s0.getIntValues()) s0_shape.push_back(s.getSExtValue()); in ExtractInputConstShape()
598 for (auto s : s1.getIntValues()) s1_shape.push_back(s.getSExtValue()); in ExtractInputConstShape()
2786 llvm::transform(reductions_value.getIntValues(), in FoldOperandsPermutation()
|
| D | tf_ops_helpers.inc | 350 for (const auto &value : perm0.getIntValues())
354 for (const auto &value : perm1.getIntValues())
|
| D | tf_ops_n_z.cc | 320 for (auto index_pair : llvm::enumerate(paddings_value.getIntValues())) { in FoldOperandsPermutation()
609 for (const auto &dim : llvm::enumerate(shape_attr.getIntValues())) { in GetReshapeOutputType()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | BuiltinAttributes.h | 912 llvm::iterator_range<IntElementIterator> getIntValues() const;
916 return getIntValues();
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 235 for (auto size : sizeAttr.getIntValues()) in addAsyncDependency()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Operation.cpp | 1039 if (llvm::any_of(sizeAttr.getIntValues(), [](const APInt &element) { in verifyValueSizeAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 736 for (const auto& dim : llvm::enumerate(shape_attr.getIntValues())) { in ConvertOp()
|