Home
last modified time | relevance | path

Searched refs:getI64IntegerAttr (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc59 builder->getI64IntegerAttr(dnums.index_vector_dim()), in ConvertGatherDimensionNumbers()
76 builder->getI64IntegerAttr(dnums.index_vector_dim()), in ConvertScatterDimensionNumbers()
116 builder->getI64IntegerAttr(dnums.input_batch_dimension()), in ConvertConvDimensionNumbers()
117 builder->getI64IntegerAttr(dnums.input_feature_dimension()), in ConvertConvDimensionNumbers()
119 builder->getI64IntegerAttr(dnums.kernel_input_feature_dimension()), in ConvertConvDimensionNumbers()
120 builder->getI64IntegerAttr(dnums.kernel_output_feature_dimension()), in ConvertConvDimensionNumbers()
122 builder->getI64IntegerAttr(dnums.output_batch_dimension()), in ConvertConvDimensionNumbers()
123 builder->getI64IntegerAttr(dnums.output_feature_dimension()), in ConvertConvDimensionNumbers()
Dhlo_function_importer.cc254 func_builder->getI64IntegerAttr( in ImportInstructionImpl()
279 builder_->getI64IntegerAttr(instruction->feature_index()))); \ in ImportInstructionImpl()
474 "dimension", builder_->getI64IntegerAttr(instruction->dimension()))); in ImportInstructionImpl()
496 builder_->getI64IntegerAttr(sort_instruction->sort_dimension()), in ImportInstructionImpl()
551 builder_->getI64IntegerAttr(instruction->concatenate_dimension())) in ImportInstructionImpl()
630 "dimension", builder_->getI64IntegerAttr(instruction->dimension()))); in ImportInstructionImpl()
714 builder_->getI64IntegerAttr(instruction->feature_group_count()))); in ImportInstructionImpl()
717 builder_->getI64IntegerAttr(instruction->batch_group_count()))); in ImportInstructionImpl()
986 builder_->getI64IntegerAttr(channel.handle()), in ConvertChannelHandle()
987 builder_->getI64IntegerAttr(channel.type()), context_)); in ConvertChannelHandle()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_gather_to_torch_index_select.cc120 operand, gather.start_indices(), rewriter.getI64IntegerAttr(0), in matchAndRewrite()
121 rewriter.getI64IntegerAttr(0)); in matchAndRewrite()
Dlhlo_legalize_to_gpu.cc128 auto oneAttr = rewriter.getI64IntegerAttr(1); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_attr.cc35 return builder->getI64IntegerAttr(value.i()); in ConvertNonFuncAttributeValue()
54 attrs.push_back(builder->getI64IntegerAttr(item)); in ConvertNonFuncAttributeValue()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc420 sort.dimensionAttr(builder_.getI64IntegerAttr(sort_instr->sort_dimension())); in EmitSortOp()
553 builder_.getI64IntegerAttr(xla_scatter_dim.index_vector_dim()), in GetScatterDimensionNumbers()
677 op.batch_sizeAttr(builder_.getI64IntegerAttr(config.batch_size())); in EmitGemm()
680 op.algorithmAttr(builder_.getI64IntegerAttr(config.selected_algorithm())); in EmitGemm()
773 builder_.getI64IntegerAttr(custom_call->feature_group_count())); in EmitDnnConvolution()
775 builder_.getI64IntegerAttr(custom_call->batch_group_count())); in EmitDnnConvolution()
781 builder_.getI64IntegerAttr(backend_config.algorithm()), in EmitDnnConvolution()
862 op.feature_indexAttr(builder_.getI64IntegerAttr(feature_index_value)); in EmitDnnBatchNorm()
1029 builder.getI64IntegerAttr(*instr->channel_id()), in SetupChannelIdAttribute()
1030 builder.getI64IntegerAttr(0), builder.getContext())); in SetupChannelIdAttribute()
[all …]
Dlegalize_tf.cc209 return b->getI64IntegerAttr(axis); in GetHLOAxisFromTFAxis()
735 return b.getI64IntegerAttr( in getFeatureDimensionAttr()
1097 builder->getI64IntegerAttr(GetTensorBatchDimIndex(num_dims, format)); in GetConvDimensionNumbersAttr()
1099 builder->getI64IntegerAttr(GetTensorFeatureDimIndex(num_dims, format)); in GetConvDimensionNumbersAttr()
1106 builder->getI64IntegerAttr(num_spatial_dims); in GetConvDimensionNumbersAttr()
1108 builder->getI64IntegerAttr(num_spatial_dims + 1); in GetConvDimensionNumbersAttr()
1251 "feature_group_count", rewriter.getI64IntegerAttr(feature_group_count)); in matchAndRewrite()
1254 "batch_group_count", rewriter.getI64IntegerAttr(1)); in matchAndRewrite()
1397 rewriter.getI64IntegerAttr(0)); in matchAndRewrite()
1399 rewriter.getI64IntegerAttr(1)); in matchAndRewrite()
[all …]
Dlegalize_tf_communication.cc259 /*handle=*/builder.getI64IntegerAttr(channel_id++), in CreateSendOp()
260 /*type=*/builder.getI64IntegerAttr(2), builder.getContext()); in CreateSendOp()
280 /*handle=*/builder.getI64IntegerAttr(channel_id++), in CreateRecvOp()
281 /*type=*/builder.getI64IntegerAttr(3), builder.getContext()); in CreateRecvOp()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc199 IntegerAttr concat_axis_attr = rewriter.getI64IntegerAttr(concat_axis); in convertPackOp()
689 rewriter.getI64IntegerAttr(axis)); in convertConcatV2Op()
724 rewriter.getI64IntegerAttr(axis)); in convertConcatV2Op()
1457 rewriter.getI64IntegerAttr(input_rank - 1)); in convertSoftmaxOp()
1495 rewriter.getI64IntegerAttr(input_rank - 1)); in convertSoftmaxOp()
1592 rewriter.getI64IntegerAttr(input_rank - 1)); in convertSoftmaxOp()
1653 rewriter.getI64IntegerAttr(input_rank - 1)); in convertLogSoftmaxOp()
2235 rewriter.getI64IntegerAttr(std::numeric_limits<int32_t>::max()), in convertFusedActivation()
2246 rewriter.getI64IntegerAttr(rescaled_6), in convertFusedActivation()
2259 rewriter.getI64IntegerAttr(rescaled_n1), in convertFusedActivation()
[all …]
Dlegalize_tf.cc148 op, output_type, tf_relu_op.features(), rewriter.getI64IntegerAttr(0), in matchAndRewrite()
153 rewriter.getI64IntegerAttr(std::numeric_limits<int32_t>::max()), in matchAndRewrite()
170 op, output_type, tf_relu6_op.features(), rewriter.getI64IntegerAttr(0), in matchAndRewrite()
174 op, output_type, tf_relu6_op.features(), rewriter.getI64IntegerAttr(6), in matchAndRewrite()
486 IntegerAttr axis_attr = rewriter.getI64IntegerAttr(axis); in matchAndRewrite()
1501 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()
1523 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()
1757 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()
1961 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
Dlegalize_tfl.cc180 rewriter.getI64IntegerAttr(std::numeric_limits<int32_t>::max()), in matchAndRewrite()
190 rewriter.getI64IntegerAttr(0), in matchAndRewrite()
239 rewriter.getI64IntegerAttr(rescaled_6), rewriter.getF32FloatAttr(0.0f)); in matchAndRewrite()
248 rewriter.getI64IntegerAttr(0), rewriter.getF32FloatAttr(6.0f)); in matchAndRewrite()
1559 tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()
2010 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()
2032 if (!tmpAttr) tmpAttr = rewriter.getI64IntegerAttr(0); in matchAndRewrite()
2534 rewriter.getI64IntegerAttr(0), rewriter.getF32FloatAttr(6.0)); in matchAndRewrite()
2775 auto axis_attr = rewriter.getI64IntegerAttr(axis_val); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc175 builder->getI64IntegerAttr(begin_mask), in CreateStridedSliceOp()
176 builder->getI64IntegerAttr(end_mask), in CreateStridedSliceOp()
177 builder->getI64IntegerAttr(ellipsis_mask), in CreateStridedSliceOp()
178 builder->getI64IntegerAttr(new_axis_mask), in CreateStridedSliceOp()
179 builder->getI64IntegerAttr(shrink_axis_mask)); in CreateStridedSliceOp()
Dperception_ops_utils_test.cc77 ret.push_back(builder->getI64IntegerAttr(value)); in createInt64Array()
/external/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
DLLVMDialect.cpp100 parser.getBuilder().getI64IntegerAttr(predicateValue)); in parseCmpOp()
206 result.addAttribute("alignment", builder.getI64IntegerAttr(alignment)); in build()
267 result.addAttribute("alignment", builder.getI64IntegerAttr(alignment)); in build()
1036 builder.getI64IntegerAttr(static_cast<int64_t>(linkage))); in build()
1141 result.addAttribute(name, parser.getBuilder().getI64IntegerAttr(index)); in parseOptionalLLVMKeyword()
1154 parser.getBuilder().getI64IntegerAttr( in parseGlobalOp()
1312 builder.getI64IntegerAttr(static_cast<int64_t>(linkage))); in build()
1374 parser.getBuilder().getI64IntegerAttr( in parseLLVMFuncOp()
1561 auto attr = parser.getBuilder().getI64IntegerAttr(value); in parseAtomicBinOp()
1588 auto attr = parser.getBuilder().getI64IntegerAttr(value); in parseAtomicOrdering()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_outside_compilation.cc105 /*device_ordinal=*/builder.getI64IntegerAttr(0)); in CreateSendFromHostOp()
122 /*device_ordinal=*/builder.getI64IntegerAttr(0)); in CreateRecvAtHostOp()
249 /*tpu_core=*/builder.getI64IntegerAttr(0)); in CreateHostCompute()
Dreplicate_to_island.cc99 op->setAttr(kReplicaIdAttr, builder.getI64IntegerAttr(replica_id)); in UpdateRegionReplicateVariantOps()
Dtpu_dynamic_layout_pass.cc152 builder->getI64IntegerAttr(execute_arg_index)), in BuildGetLayout()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc114 builder_.getI64IntegerAttr(feature_group_count), in ConvGeneralDilatedInternal()
115 builder_.getI64IntegerAttr(batch_group_count), config_attr); in ConvGeneralDilatedInternal()
253 builder_.getI64IntegerAttr(dimension), builder_.getBoolAttr(is_stable)); in SortInternal()
326 builder_.getI64IntegerAttr(dimension)); in SetDimensionSizeInternal()
494 loc_, result_type, mlir_operands, builder_.getI64IntegerAttr(dimension))); in ConcatInDimInternal()
/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/
DParallelLoopMapper.cpp39 builder.getI64IntegerAttr(static_cast<int32_t>(processor)), in getParallelLoopDimMappingAttr()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/
Dtf_to_quant.cc110 IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.num_bits()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DGPUOpsLowering.h135 rewriter.getI64IntegerAttr(type.getNumElements())); in matchAndRewrite()
/external/llvm-project/mlir/lib/IR/
DBuilders.cpp102 IntegerAttr Builder::getI64IntegerAttr(int64_t value) { in getI64IntegerAttr() function in Builder
234 values, [this](int64_t v) -> Attribute { return getI64IntegerAttr(v); })); in getI64ArrayAttr()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dimport_quant_stats_pass.cc158 axis = b.getI64IntegerAttr(info.meta().quantize_axis()); in ImportAsStatsOps()
/external/llvm-project/mlir/lib/Transforms/
DPipelineDataTransfer.cpp354 op.setAttr("shift", b.getI64IntegerAttr(shifts[s - 1])); in runOnAffineForOp()
/external/llvm-project/flang/lib/Lower/
DFIRBuilder.cpp180 mlir::NamedAttribute sizeAttr(sizeTag, getI64IntegerAttr(data.size())); in createStringLit()

123