| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | promote_resources_to_args.cc | 128 << resource_arg.getArgNumber() << " to have one subtype, got " in ValidateResourceArgument()
135 << resource_arg.getArgNumber() in ValidateResourceArgument()
206 resources[arg.getArgNumber()].live_value = arg; in PromoteResourcesToArguments()
207 argument_types[arg.getArgNumber()] = arg_type; in PromoteResourcesToArguments()
251 ResourceInfo& resource_info = resources[func_arg.getArgNumber()]; in PromoteResourcesToArguments()
264 ResourceInfo& resource_info = resources[func_arg.getArgNumber()]; in PromoteResourcesToArguments()
|
| D | resource_device_inference.cc | 154 StringRef device_attr = GetDeviceAttr(func_op, arg.getArgNumber()); in ComputeResourceDevicesInComputation()
159 func_op.setArgAttr(arg.getArgNumber(), kFuncDeviceAttr, in ComputeResourceDevicesInComputation()
255 Value arg_operand = caller_operands[arg.getArgNumber()]; in runOnOperation()
260 << arg.getArgNumber() << " of function @" in runOnOperation()
|
| D | resource_op_lifting_cleanup.cc | 168 if (!arg.use_empty()) used_args.set(arg.getArgNumber()); in EliminateUnusedResultsForIfCase()
268 common_arg_index = block_arg.getArgNumber(); in ForwardCommonArgToOutput()
269 } else if (common_arg_index.getValue() != block_arg.getArgNumber()) { in ForwardCommonArgToOutput()
384 if (!body_arg || body_arg.getArgNumber() != result_idx) { in CanonicalizeWhileRegion()
|
| D | resource_op_lifting.cc | 712 (*result)[arg.getArgNumber()] = info; in FindResourceArgUseInfo()
760 auto it = infos.find(arg.getArgNumber()); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
770 auto it = infos.find(arg.getArgNumber()); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
772 indices_to_erase.push_back(arg.getArgNumber()); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
780 old_to_new_arg_indices->push_back(arg.getArgNumber() - skipped_args); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
783 (*remaining_resource_data_types)[arg.getArgNumber() - skipped_args] = in RemoveUnusedResourceArgumentsAndForwardedRetvals()
817 auto type_iter = resource_data_types.find(arg.getArgNumber()); in LiftArgRetResourcesForFunction()
841 handle_updated_arg_value(arg.getArgNumber(), assign_variable_op.value()); in LiftArgRetResourcesForFunction()
1095 aliasing_arg.getArgNumber(); in HandlePartitionedCallOpCallee()
|
| D | tpu_variable_runtime_reformatting.cc | 167 assert(replicate_arg_to_execute_arg.count(block_arg.getArgNumber()) == 0 && in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
169 replicate_arg_to_execute_arg[block_arg.getArgNumber()] = in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
316 new_replicate.GetBody().getArgument(arg.getArgNumber())); in AddInputsToReplicateOp()
321 new_replicate.GetBody().getArgument(arg.getArgNumber() + 1)); in AddInputsToReplicateOp()
|
| D | tensor_device_copy_conversion.cc | 54 block_arg.getArgNumber(), kTFDeviceAttr)) { in runOnFunction()
|
| D | annotate_parameter_replication.cc | 78 if (mirrored_replicate_args.count(block_arg.getArgNumber()) == 0) { in runOnOperation()
|
| D | test_resource_alias_analysis.cc | 90 << region_arg.getArgNumber(); in runOnFunction()
|
| D | stack_ops_decomposition.cc | 105 return maybe_arg.getArgNumber(); in FindAliasedInput()
357 [entry.getFirst().cast<BlockArgument>().getArgNumber()] = in HandlePartitionedCallOp()
358 entry.getSecond().cast<BlockArgument>().getArgNumber(); in HandlePartitionedCallOp()
|
| D | cross_host_transfer.cc | 102 block_arg.getArgNumber(), kArgDeviceAttr)) { in runOnFunction()
|
| D | region_control_flow_to_functional.cc | 374 return first_block_arg.getArgNumber() == second_block_arg.getArgNumber() && in ConvertWhileOp()
|
| D | tpu_dynamic_padding_mapper.cc | 68 replicated_input_indices_attr[block_arg.getArgNumber()] in GetRemappedReplicatedInputIndices()
|
| D | lift_variables.cc | 187 unsigned arg_number = arg.getArgNumber(); in LiftVariables()
|
| D | tensor_array_ops_decomposition.cc | 477 auto insert_res = result_sets[arg.getArgNumber()].insert(source); in AccessedGradients()
479 result[arg.getArgNumber()].push_back(source); in AccessedGradients()
700 return arg.getArgNumber(); in HandleIfOp()
809 info.ret_forward_input.emplace_back(i, arg.getArgNumber()); in HandlePartitionedCallOp()
|
| D | tpu_device_propagation.cc | 144 func.getArgAttrOfType<StringAttr>(arg.getArgNumber(), kFuncDeviceAttr); in PropagateDevicesFromArguments()
|
| D | tpu_sharding_identification_pass.cc | 149 const int index = arg.getArgNumber(); in IdentifyXlaShardingForComputationInputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 389 const int block_arg_num = arg.getArgNumber(); in Print()
476 << ") and block argument " << block_arg.getArgNumber() << " (" in Verify()
482 if (block_arg.getArgNumber() < num_replicated_block_args) { in Verify()
483 for (int32_t i = n * block_arg.getArgNumber(), e = i + n; i < e; ++i) in Verify()
486 const int32_t idx = block_arg.getArgNumber() - num_replicated_block_args + in Verify()
614 return block_arg.getArgNumber() < GetNumReplicatedBlockArguments(); in IsReplicatedBlockArgument()
653 unsigned arg_number = block_arg.getArgNumber(); in GetOperandsForBlockArgument()
|
| /external/llvm-project/mlir/test/lib/IR/ |
| D | TestPrintDefUse.cpp | 36 << blockArg.getArgNumber() << "\n"; in runOnOperation()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestAffineLoopParametricTiling.cpp | 52 if (blockArgument.getArgNumber() < nestDepth) { in getTilingParameters()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 75 if (arg.getParentBlock() == ®ion_->front()) return arg.getArgNumber(); in GetArg()
270 return func_op.getArgAttr(arg.getArgNumber(), kResourceArgUniqueIdAttr); in ResourceAliasAnalysisInfo()
278 arg.getArgNumber(), kResourceArgUniqueIdAttr); in ResourceAliasAnalysisInfo()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | BufferDeallocation.cpp | 260 .getValue()[blockArg.getArgNumber()]; in introduceBlockArgCopy()
271 .slice(blockArg.getArgNumber(), 1) in introduceBlockArgCopy()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | op_or_arg_name_mapper.cc | 130 return "arg" + std::to_string(arg.getArgNumber()); in GetName()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_variables.cc | 61 int index = block_arg.getArgNumber(); in GetGlobalTensor()
|
| D | initialize_variables.cc | 55 int index = block_arg.getArgNumber(); in GetGlobalTensor()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Block.cpp | 20 unsigned BlockArgument::getArgNumber() const { in getArgNumber() function in BlockArgument
|