Home
last modified time | relevance | path

Searched refs:getAttrOfType (Results 1 – 25 of 116) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dvalidators.h34 auto attr = op->getAttrOfType<StringAttr>("data_format"); in TFDataFormatIsNHWC()
42 auto attr = op->getAttrOfType<StringAttr>("data_format"); in TFDataFormatIsNDHWC()
97 auto padding_attr = op->getAttrOfType<StringAttr>("padding"); in TFPaddingIsSameOrValid()
Dvalidators.cc30 auto attr = op->getAttrOfType<ArrayAttr>(name); in TFIntListIs1XY1()
70 auto attr = op->getAttrOfType<ArrayAttr>(name); in TFIntListIs1XYZ1()
Dlstm_utils_test.cc129 fused_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), in TEST_F()
202 EXPECT_EQ(fused_lstm_func_cifg_->getAttrOfType<StringAttr>(kTFImplements) in TEST_F()
227 fused_ln_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtest_visitor_util.cc67 op->getAttrOfType<mlir::BoolAttr>("interrupt_before_all")) in runOnFunction()
72 op->getAttrOfType<mlir::BoolAttr>("interrupt_after_all")) in runOnFunction()
77 op->getAttrOfType<mlir::IntegerAttr>("interrupt_after_region")) in runOnFunction()
Dexecutor_tpuv1_island_coarsening.cc131 wrapped_op.getAttrOfType<StringAttr>(kTpuReplicateAttr); in MergeIsland()
133 cluster_name = wrapped_op.getAttrOfType<StringAttr>(kTpuStatusAttr); in MergeIsland()
151 candidate_wrapped_op.getAttrOfType<StringAttr>(kTpuReplicateAttr); in MergeIsland()
154 candidate_wrapped_op.getAttrOfType<StringAttr>(kTpuStatusAttr); in MergeIsland()
271 op->getAttrOfType<StringAttr>(kTpuReplicateAttr); in runOnOperation()
273 cluster_name = op->getAttrOfType<StringAttr>(kTpuStatusAttr); in runOnOperation()
Dmark_ops_for_outside_compilation.cc218 if (parent->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in HasOutsideCompiledAncestor()
259 !user->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in MarkVariantInputsOutputs()
307 if (!op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) return; in UnmarkChildren()
311 if (parent_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in UnmarkChildren()
352 cluster->getAttrOfType<BoolAttr>(kAllowSoftPlacementAttr); in runOnOperation()
369 cluster->getAttrOfType<BoolAttr>(kAllowSoftPlacementAttr); in runOnOperation()
Dreadonly_references_to_resources.cc66 ArrayAttr classes_attr = op->getAttrOfType<ArrayAttr>(kClassAttr); in GetNodeNameFromClassAttr()
72 classes_attr = identity_op->getAttrOfType<ArrayAttr>(kClassAttr); in GetNodeNameFromClassAttr()
142 variable_v2_op->getAttrOfType<StringAttr>("device"); in runOnFunction()
Dtensor_device_copy_conversion.cc58 arg.getDefiningOp()->getAttrOfType<StringAttr>( in runOnFunction()
63 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnFunction()
Dconstant_op_device_assignment.cc45 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation()
55 StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation()
Dtpu_rewrite_pass.cc153 op->getAttrOfType<StringAttr>(kStepMarkerLocationAttr); in SetMetadataProtoStepMarkerLocation()
178 auto padding_map = op->getAttrOfType<ArrayAttr>(kPaddingMapAttr); in SetMetadataProtoPaddingMap()
222 op->getAttrOfType<ArrayAttr>(tensorflow::kInputShardingAttr); in SetMetadataProtoArgs()
277 op->getAttrOfType<ArrayAttr>(tensorflow::kOutputShardingAttr); in SetMetadataProtoRetvals()
317 auto use_spmd_attr = op->getAttrOfType<BoolAttr>(kUseXlaSpmdAttr); in SetMetadataProtoFromClusterFuncOp()
576 auto num_cores_per_replica_attr = cluster_func->getAttrOfType<IntegerAttr>( in Rewrite()
585 cluster_func->getAttrOfType<StringAttr>(tensorflow::kTopologyAttr); in Rewrite()
590 auto device_assignment_attr = cluster_func->getAttrOfType<mlir::ArrayAttr>( in Rewrite()
740 auto replicate_attr = op->getAttrOfType<StringAttr>("_tpu_replicate"); in runOnOperation()
Dtpu_device_propagation.cc63 auto lhs_device_attr = lhs->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedExecutorOp()
64 auto rhs_device_attr = rhs->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedExecutorOp()
175 auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); in PropagateDevicesInGraph()
Dtpu_update_embedding_enqueue_op_inputs.cc51 auto embedding_attr = op->getAttrOfType<StringAttr>(kTPUEmbeddingAttr); in ExtractEmbeddingAttribute()
129 embedding_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr); in UpdateEmbeddingEnqueueOpInput()
Dcross_host_transfer.cc88 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction()
108 defining_op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction()
Dtf_device_assignment.cc39 if (auto device_attr = op->getAttrOfType<StringAttr>("device")) { in runOnFunction()
/external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/
DConvertLaunchFuncToLLVMCalls.cpp55 IntegerAttr binding = op->getAttrOfType<IntegerAttr>(bindingName()); in calculateGlobalIndex()
78 op->getAttrOfType<IntegerAttr>(descriptorSetName()); in createGlobalVariableWithBindName()
79 IntegerAttr binding = op->getAttrOfType<IntegerAttr>(bindingName()); in createGlobalVariableWithBindName()
90 op->getAttrOfType<IntegerAttr>(descriptorSetName()); in hasDescriptorSetAndBinding()
91 IntegerAttr binding = op->getAttrOfType<IntegerAttr>(bindingName()); in hasDescriptorSetAndBinding()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Ddevice_target.cc134 auto min = rop->getAttrOfType<FloatAttr>("min"); in DecomposeMultiplyAccumulateScale()
135 auto max = rop->getAttrOfType<FloatAttr>("max"); in DecomposeMultiplyAccumulateScale()
169 auto min = rop->getAttrOfType<FloatAttr>("min"); in DecomposeSameScale()
170 auto max = rop->getAttrOfType<FloatAttr>("max"); in DecomposeSameScale()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtranslate_utils.cc42 module->getAttrOfType<mlir::DictionaryAttr>("tf.versions"); in ExtractTfVersions()
70 auto versions = module->getAttrOfType<::mlir::DictionaryAttr>("tf.versions"); in GetTfGraphProducerVersion()
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/
DOps.h63 return (*this)->getAttrOfType<FloatAttr>("value").getValue(); in getValue()
87 return (*this)->getAttrOfType<IntegerAttr>("value").getInt(); in getValue()
106 return (*this)->getAttrOfType<IntegerAttr>("value").getInt(); in getValue()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc345 func->getAttrOfType<StringAttr>(kTFImplements); in runOnOperation()
351 auto tf_implements_attr = func->getAttrOfType<FuncAttr>(kTFImplements); in runOnOperation()
358 func->getAttrOfType<StringAttr>(kTFAPIImplements); in runOnOperation()
/external/llvm-project/flang/lib/Optimizer/Dialect/
DFIROps.cpp243 op.template getAttrOfType<mlir::IntegerAttr>(OPTY::getPredicateAttrName()) in printCmpOp()
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()
1076 getAttrOfType<mlir::DenseIntElementsAttr>(getOperandSegmentSizeAttr()); in getSuccessorOperands()
1156 getAttrOfType<mlir::DenseIntElementsAttr>(getCompareOffsetAttr())); in compareOffsetSize()
[all …]
/external/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
DNVVMDialect.cpp112 auto alayout = op->getAttrOfType<StringAttr>("alayout"); in verify()
113 auto blayout = op->getAttrOfType<StringAttr>("blayout"); in verify()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_tf_framework.cc79 auto reuse_input_candidates = alloc->getAttrOfType<ArrayAttr>( in matchAndRewrite()
82 alloc->getAttrOfType<IntegerAttr>(TFAllocOp::kReuseOutputAttrName); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/
DLowerABIAttributesPass.cpp111 funcOp->getAttrOfType<spirv::EntryPointABIAttr>(entryPointAttrName); in lowerEntryPointABIAttr()
172 if (!funcOp->getAttrOfType<spirv::EntryPointABIAttr>( in matchAndRewrite()
274 if (funcOp->getAttrOfType<spirv::EntryPointABIAttr>(entryPointAttrName)) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc54 func->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function")) { in HandleInputOutputArraysWithModule()
70 entry_function->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); in HandleInputOutputArraysWithModule()
/external/llvm-project/mlir/lib/Dialect/GPU/IR/
DGPUDialect.cpp35 UnitAttr isKernelAttr = op->getAttrOfType<UnitAttr>(getKernelFuncAttrName()); in isKernel()
89 if (!launchOp->getAttrOfType<SymbolRefAttr>( in verifyOperationAttribute()
107 if (!kernelFunc->getAttrOfType<mlir::UnitAttr>( in verifyOperationAttribute()
231 auto sizeAttr = op->template getAttrOfType<DenseIntElementsAttr>(attrName); in addAsyncDependency()
477 if (!module->getAttrOfType<UnitAttr>( in verify()
483 auto kernelAttr = op->getAttrOfType<SymbolRefAttr>(op.getKernelAttrName()); in verify()
526 auto attr = (*this)->getAttrOfType<IntegerAttr>(attrName); in addWorkgroupAttribution()

12345