Home
last modified time | relevance | path

Searched refs:getOpOperands (Results 1 – 25 of 29) sorted by relevance

12

/external/llvm-project/mlir/lib/Transforms/Utils/
DFoldUtils.cpp175 op->getOpOperands().begin(), op->getOpOperands().end(), in tryToFold()
DInliningUtils.cpp48 for (auto &operand : op->getOpOperands()) in remapInlinedOperands()
DRegionUtils.cpp45 for (OpOperand &operand : op->getOpOperands()) in visitUsedValuesDefinedAbove()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dremove_vars_in_session_initializer.cc54 for (auto& use : op->getOpOperands()) { in RecursiveRemove()
Dtpu_reorder_replicate_and_partitioned_inputs.cc84 for (auto& pi_operand : pi->getOpOperands()) { in ReorderReplicateAndPartitionedInputs()
Dtpu_resource_partitioning.cc115 for (OpOperand& operand : cluster_func->getOpOperands()) { in PartitionResourceReadsWrites()
Dtpu_device_propagation.cc214 for (OpOperand& operand : fetch.getOperation()->getOpOperands()) { in PropagateDevicesToResults()
Dtpu_sharding_identification_pass.cc264 llvm::zip(cluster_func.results(), terminator->getOpOperands())) { in IdentifyXlaShardingForComputationOutputs()
Dlayout_optimization.cc273 for (OpOperand& operand : op->getOpOperands()) { in MoveTransposeBefore()
Dtpu_merge_variables_with_execute.cc145 for (auto operand : llvm::enumerate(execute->getOpOperands())) { in BuildVariableAccessInfo()
Dstack_ops_decomposition.cc471 for (OpOperand& operand : op.getOpOperands()) { in HandleRegionControlFlowOps()
Dtensor_array_ops_decomposition.cc834 for (OpOperand& operand : op.getOpOperands()) { in HandleRegionControlFlowOps()
Dresource_op_lifting.cc1128 for (auto& val : callee.front().getTerminator()->getOpOperands()) { in HandlePartitionedCallOpCallee()
Dshape_inference.cc1479 for (OpOperand& arg_op : return_op.getOperation()->getOpOperands()) { in InferShapeForFunctionReturnType()
/external/llvm-project/mlir/docs/Rationale/
DUsageOfConst.md147 ArrayRef<OpOperand> getOpOperands() const {
150 MutableArrayRef<OpOperand> getOpOperands() {
154 OpOperand &getOpOperand(unsigned idx) { return getOpOperands()[idx]; }
156 return getOpOperands()[idx];
/external/llvm-project/mlir/lib/IR/
DValue.cpp242 return this - &getOwner()->getOpOperands()[0]; in getOperandNumber()
DRegion.cpp109 for (auto &operand : op->getOpOperands()) in cloneInto()
DOperation.cpp248 for (auto &operand : getOpOperands()) in replaceUsesOfWith()
533 for (auto &op : getOpOperands()) in dropAllReferences()
DOperationSupport.cpp404 : OperandRange(op->getOpOperands().data(), op->getNumOperands()) {} in OperandRange()
/external/llvm-project/mlir/include/mlir/IR/
DOperation.h256 MutableArrayRef<OpOperand> getOpOperands() { in getOpOperands() function
261 OpOperand &getOpOperand(unsigned idx) { return getOpOperands()[idx]; } in getOpOperand()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc656 MutableArrayRef<OpOperand> operands = getOperation()->getOpOperands(); in GetOperandsForBlockArgument()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1310 for (OpOperand& ret : inst->getOpOperands()) { in Lower()
1811 for (OpOperand& ret : inst.getOpOperands()) { in BuildHloFromMlirHlo()
/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp135 for (OpOperand &operand : op->getOpOperands()) { in foldMemRefCast()
/external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/
DDeserializer.cpp2083 for (auto &operand : op->getOpOperands()) in structurizeImpl()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp2480 for (OpOperand &operand : definedOp->getOpOperands()) in matchAndRewrite()

12