| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 180 body.getArgument(use.getOperandNumber()), module, infer_from_op); in GetElementTypeFromAccess()
187 then_branch.getArgument(use.getOperandNumber() - 1), module, in GetElementTypeFromAccess()
191 else_branch.getArgument(use.getOperandNumber() - 1), module, in GetElementTypeFromAccess()
197 callee.getArgument(use.getOperandNumber()), module, infer_from_op); in GetElementTypeFromAccess()
201 use.getOwner()->getResult(use.getOperandNumber()), module, in GetElementTypeFromAccess()
|
| D | constant_op_device_assignment.cc | 70 user_op->setOperand(use.getOperandNumber(), new_op->getResult(0)); in runOnOperation()
|
| D | tpu_device_propagation.cc | 219 operand.getOperandNumber(), kFuncDeviceAttr); in PropagateDevicesToResults()
221 func.setResultAttr(operand.getOperandNumber(), kFuncDeviceAttr, in PropagateDevicesToResults()
|
| D | tpu_sharding_identification_pass.cc | 103 func.getArgument(use.getOperandNumber())); in GetXlaShardingFromArg()
267 const int index = retval.getOperandNumber(); in IdentifyXlaShardingForComputationOutputs()
|
| D | tpu_reorder_replicate_and_partitioned_inputs.cc | 85 unsigned core_id = pi_operand.getOperandNumber(); in ReorderReplicateAndPartitionedInputs()
|
| D | tpu_cluster_formation.cc | 476 pi.setOperand(use.getOperandNumber(), block_arg); in ReplicateCluster()
615 << op->use_begin()->getOperandNumber() << " of " in FormClustersInFunction()
|
| D | resource_op_lifting_cleanup.cc | 210 body_arg.use_begin()->getOperandNumber() == result_idx && in EliminateUnusedResultsForWhile()
|
| D | resource_op_lifting.cc | 392 info.result_index = use.getOperandNumber(); in Analyze()
1131 retval_indices_to_preserve.push_back(val.getOperandNumber()); in HandlePartitionedCallOpCallee()
|
| D | stack_ops_decomposition.cc | 475 << " of operand #" << operand.getOperandNumber() in HandleRegionControlFlowOps()
|
| D | shape_inference.cc | 245 branch->getArgument(use.getOperandNumber()), in CanInferTensorListElementType()
254 parent->getResult(use.getOperandNumber()), initial_element_shape, in CanInferTensorListElementType()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Value.cpp | 217 unsigned BlockOperand::getOperandNumber() { in getOperandNumber() function in BlockOperand
241 unsigned OpOperand::getOperandNumber() { in getOperandNumber() function in OpOperand
|
| D | Block.cpp | 298 return I->getOperandNumber(); in getSuccessorIndex()
|
| D | OperationSupport.cpp | 410 return base->getOperandNumber(); in getBeginOperandIndex()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | UseDefLists.h | 221 unsigned getOperandNumber();
262 unsigned getOperandNumber();
|
| /external/llvm-project/mlir/lib/Conversion/PDLToPDLInterp/ |
| D | Predicate.h | 194 unsigned getOperandNumber() const { return key.second; } in getOperandNumber() function
438 index.push_back(p->getOperandNumber()); in getParent()
|
| D | PDLToPDLInterp.cpp | 239 operandPos->getOperandNumber()); in getValueAt()
588 if (!replOpUser || use.getOperandNumber() == 0) in generateOperationResultTypeRewriter()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | default_quant_params.cc | 65 if (biases.find(use.getOperandNumber()) != biases.end()) return true; in UsedAsBias()
|
| D | lower_static_tensor_list.cc | 363 .getArgument(use.getOperandNumber()) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 628 if (operands.size() == 1) return operands.front().getOperandNumber(); in GetReplicaOperandIndexForBlockArgument()
630 return operands[replica].getOperandNumber(); in GetReplicaOperandIndexForBlockArgument()
|
| D | tf_executor.cc | 1042 use.getOwner()->eraseOperand(use.getOperandNumber()); in matchAndRewrite()
1097 use.getOwner()->eraseOperand(use.getOperandNumber()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | buffer_reuse_pass.cc | 178 int32_t index = use.getOperandNumber(); in find_output_indices()
|
| /external/llvm-project/mlir/lib/Dialect/PDL/IR/ |
| D | PDL.cpp | 227 if (!replOpUser || use.getOperandNumber() == 0) in verifyResultTypesAreInferrable()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | RegionUtils.cpp | 164 unsigned operandIndex = use.getOperandNumber(); in isUseSpeciallyKnownDead()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_driver.cc | 685 uses.push_back({use.getOwner(), use.getOperandNumber()}); in PreprocessConstantOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1311 unsigned index = ret.getOperandNumber(); in Lower()
1812 unsigned index = ret.getOperandNumber(); in BuildHloFromMlirHlo()
|