Searched refs:getI64Type (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.cc | 128 Type state_element_type = rewriter.getI64Type(); in matchAndRewrite()
195 Type i64 = rewriter.getI64Type(); in matchAndRewrite()
|
| D | legalize_hlo.cc | 480 RankedTensorType::get({lhs_rank}, rewriter.getI64Type()), in ConvertDot()
498 RankedTensorType::get({rhs_rank}, rewriter.getI64Type()), in ConvertDot()
632 {static_cast<int64_t>(reduce_dims.size())}, rewriter.getI64Type()); in matchAndRewrite()
1181 rewriter.getI64Type()); in ConvertPadOp()
|
| D | replicate_to_island.cc | 112 RankedTensorType::get({}, builder.getI64Type()), in UpdateRegionReplicateVariantOps()
|
| D | canonicalize.td | 85 "DenseElementsAttr::get(RankedTensorType::get({2}, $_builder.getI64Type()), "
|
| D | tpu_extract_outside_compilation.cc | 443 tpu_cluster.getLoc(), RankedTensorType::get({}, builder.getI64Type())); in CreateParallelExecuteForOutsideCompilation()
|
| D | lower_tf.cc | 1363 loc, RankedTensorType::get({4}, rewriter.getI64Type()), input); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_memref_prints.cc | 73 element_type = b->getI64Type(); in EmitPrint()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 55 auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); in createMaxUnpoolingFunc()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 74 IntegerType getI64Type();
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 139 result_elt_type = rewriter.getI64Type(); in CastToNonDerivedType()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 59 IntegerType Builder::getI64Type() { return IntegerType::get(64, context); } in getI64Type() function in Builder
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 208 builder_.getIntegerAttr(builder_.getI64Type(), iota_dimension)); in Iota()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVM.cpp | 130 idxType = rewriter.getI64Type(); in buildVectorComparison()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 387 rewriter.getI64Type()), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3505 op.getLoc(), RankedTensorType::get({}, rewriter.getI64Type()), ceil); in matchAndRewrite()
3507 op.getLoc(), RankedTensorType::get({1}, rewriter.getI64Type()), steps); in matchAndRewrite()
|