Searched refs:getI64TensorAttr (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | materialize_broadcasts.cc | 54 rewriter.getI64TensorAttr(operand_shape)); in matchAndRewrite()
62 rewriter.getI64TensorAttr(operand_shape)); in matchAndRewrite()
|
| D | chlo_legalize_to_hlo.cc | 76 op, result_ty, constant, shape, rewriter.getI64TensorAttr({})); in matchAndRewrite()
1112 rewriter.getI64TensorAttr(lhs_broadcast_dimensions)); in matchAndRewrite()
1120 rewriter.getI64TensorAttr(rhs_broadcast_dimensions)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1136 builder_.getI64TensorAttr(instr->dimensions())); in EmitBroadcastOp()
1163 reverse.dimensionsAttr(builder_.getI64TensorAttr(instr->dimensions())); in EmitReverseOp()
1171 transpose.permutationAttr(builder_.getI64TensorAttr(instr->dimensions())); in EmitTransposeOp()
1185 pad.edge_padding_lowAttr(builder_.getI64TensorAttr(low)); in EmitPadOp()
1186 pad.edge_padding_highAttr(builder_.getI64TensorAttr(high)); in EmitPadOp()
1187 pad.interior_paddingAttr(builder_.getI64TensorAttr(interior)); in EmitPadOp()
1206 reduce_window.window_dimensionsAttr(builder_.getI64TensorAttr(dims)); in EmitReduceWindowOp()
1208 reduce_window.window_stridesAttr(builder_.getI64TensorAttr(strides)); in EmitReduceWindowOp()
1211 reduce_window.base_dilationsAttr(builder_.getI64TensorAttr(base_dilations)); in EmitReduceWindowOp()
1215 builder_.getI64TensorAttr(window_dilations)); in EmitReduceWindowOp()
[all …]
|
| D | prepare_for_export.cc | 60 return_type, cst, b.getI64TensorAttr({})); in runOnFunction()
|
| D | mhlo_to_lhlo_with_xla.h | 163 return builder_.getI64TensorAttr( in GetI64DenseElementsAttr()
|
| D | legalize_tf.cc | 747 return builder->getI64TensorAttr({}); in GetInnerDimFromValue()
749 return builder->getI64TensorAttr(type.getShape().back()); in GetInnerDimFromValue()
1348 rewriter.getI64TensorAttr(broadcast_dimensions)); in matchAndRewrite()
1789 rewriter.getI64TensorAttr(fft_length)); in matchAndRewrite()
3543 return builder->getI64TensorAttr(axis); in ConvertAxisAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | init_text_file_to_import.cc | 104 op.getLoc(), rewriter.getI64TensorAttr(line_nums)); in matchAndRewrite()
|
| D | legalize_hlo.cc | 546 /*lhs_batching_dimensions=*/rewriter.getI64TensorAttr({}), in ConvertDotOp()
547 /*rhs_batching_dimensions=*/rewriter.getI64TensorAttr({}), in ConvertDotOp()
549 rewriter.getI64TensorAttr({lhs_rank == 1 ? 0 : 1}), in ConvertDotOp()
550 /*rhs_contracting_dimensions=*/rewriter.getI64TensorAttr({0}), in ConvertDotOp()
634 reduce_op.getLoc(), dim_type, rewriter.getI64TensorAttr(reduce_dims)); in matchAndRewrite()
747 iota_op.getLoc(), rewriter.getI64TensorAttr(reshape_shape)); in matchAndRewrite()
752 iota_op.getLoc(), rewriter.getI64TensorAttr(type.getShape())); in matchAndRewrite()
|
| D | optimize.cc | 40 auto cst_attr = rewriter.getI64TensorAttr(values); in GetI64ConstantTensor()
|
| D | lower_tf.cc | 1090 rewriter.getI64TensorAttr(reshaped_shape))); in matchAndRewrite()
1118 rewriter.getI64TensorAttr(permutation))); in matchAndRewrite()
1144 op.getLoc(), rewriter.getI64TensorAttr(reshaped_permuted_shape))); in matchAndRewrite()
1182 rewriter.getI64TensorAttr(start_indices)), in matchAndRewrite()
1184 rewriter.getI64TensorAttr(slice_sizes))); in matchAndRewrite()
1482 rewriter.create<ConstOp>(loc, rewriter.getI64TensorAttr({1})))); in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 132 DenseIntElementsAttr getI64TensorAttr(ArrayRef<int64_t> values);
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 131 DenseIntElementsAttr Builder::getI64TensorAttr(ArrayRef<int64_t> values) { in getI64TensorAttr() function in Builder
|