| /external/grpc-grpc/src/ruby/lib/grpc/generic/ |
| D | client_stub.rb | 150 return_op: false,
165 if return_op
227 return_op: false,
242 if return_op
318 return_op: false,
334 if return_op
440 return_op: false,
456 if return_op
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | xla_inline_device_ops.cc | 45 auto& return_op = cluster_op.GetBody().front(); in InlineDeviceOp() local
47 assert(mlir::isa<mlir::tf_device::ReturnOp>(return_op)); in InlineDeviceOp()
50 for (auto it : llvm::zip(cluster_op.getResults(), return_op.getOperands())) { in InlineDeviceOp()
52 old_val.replaceAllUsesExcept(new_val, &return_op); in InlineDeviceOp()
|
| D | promote_resources_to_args.cc | 175 auto return_op = in PromoteResourcesToArguments() local
177 if (!return_op) in PromoteResourcesToArguments()
259 auto return_operands = llvm::to_vector<4>(return_op.getOperands()); in PromoteResourcesToArguments()
260 auto result_types = llvm::to_vector<4>(return_op.getOperandTypes()); in PromoteResourcesToArguments()
270 OpBuilder builder(return_op); in PromoteResourcesToArguments()
323 builder.create<func::ReturnOp>(return_op.getLoc(), return_operands); in PromoteResourcesToArguments()
324 return_op.erase(); in PromoteResourcesToArguments()
|
| D | hoist_replicate_invariant_resource_writes.cc | 62 auto return_op = llvm::dyn_cast<tf_device::ReturnOp>( in MoveTailWritesAfterReplicate() local
70 return_op->insertOperands(return_op->getNumOperands(), assign.value()); in MoveTailWritesAfterReplicate()
|
| D | device_attribute_to_launch.cc | 48 auto* return_op = in WrapOpInLaunch() local
53 op->moveBefore(return_op); in WrapOpInLaunch()
|
| D | cluster_outlining.cc | 92 auto return_op = in BuildFunction() local
94 builder->setInsertionPoint(return_op); in BuildFunction()
95 ReplaceClusterReturnWithReturn(return_op, builder); in BuildFunction()
|
| D | outside_compiled_to_host_launch.cc | 49 auto* return_op = in WrapOpInLaunch() local
56 host_op->moveBefore(return_op); in WrapOpInLaunch()
|
| D | materialize_mlir_passthrough_op.cc | 92 Operation &return_op = block.front(); in runOnOperation() local
94 llvm::zip(op->getResults(), return_op.getOperands())) { in runOnOperation()
|
| D | resource_op_lifting.cc | 615 auto return_op = func_op.front().getTerminator(); in FindResourceArgUseInfo() local
623 if (user == return_op) continue; in FindResourceArgUseInfo()
686 auto return_op = func_op.front().getTerminator(); in RemoveUnusedResourceArgumentsAndForwardedRetvals() local
687 auto old_return_vals = llvm::to_vector<8>(return_op->getOperands()); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
694 return_op->eraseOperand(entry.index() - skipped_retvals++); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
723 llvm::to_vector<4>(return_op->getOperandTypes()))); in RemoveUnusedResourceArgumentsAndForwardedRetvals()
831 auto return_op = body.front().getTerminator(); in HandleWhileLoop() local
852 [&](int64_t index, Value value) { return_op->setOperand(index, value); }); in HandleWhileLoop()
872 int64_t update_index = return_op->getOperand(entry.getFirst()) == in HandleWhileLoop()
|
| D | legalize_hlo.cc | 78 auto return_op = llvm::dyn_cast_or_null<mhlo::ReturnOp>(terminator); in ReplaceReturnOp() local
79 if (return_op == nullptr) continue; in ReplaceReturnOp()
81 rewriter.setInsertionPoint(return_op); in ReplaceReturnOp()
82 rewriter.replaceOpWithNewOp<TF::YieldOp>(return_op, in ReplaceReturnOp()
83 return_op->getOperands()); in ReplaceReturnOp()
1547 auto return_op = dyn_cast_or_null<mhlo::ReturnOp>(&operations.back()); in MatchTopKComparator() local
1548 if (!compare_op || !return_op) return false; in MatchTopKComparator()
1555 return return_op.getOperands().front() == compare_op.getResult(); in MatchTopKComparator()
1798 mhlo::ReturnOp return_op = dyn_cast<mhlo::ReturnOp>(body.back()); in MatchBinaryReduceFunction() local
1799 if (!return_op) return failure(); in MatchBinaryReduceFunction()
[all …]
|
| D | convert_control_to_data_outputs.cc | 231 func::ReturnOp return_op = cast<func::ReturnOp>(block.getTerminator()); in AppendFunctionResults() local
232 int num_old_arguments = return_op.getNumOperands(); in AppendFunctionResults()
234 return_op.operandsMutable().append( in AppendFunctionResults()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_functional_to_executor.cc | 69 auto return_op = dyn_cast<func::ReturnOp>(body.getTerminator()); in runOnOperation() local
70 if (!return_op) { in runOnOperation()
96 builder.create<tf_executor::YieldOp>(loc, return_op.getOperands()); in runOnOperation()
98 return_op.setOperand(item.index(), item.value()); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | prepare_for_export.cc | 109 auto return_op = cast<mhlo::ReturnOp>(body_region.front().back()); in prepareWhileOp() local
110 return_op->insertOperands(return_op->getNumOperands(), body_arg); in prepareWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | reorder_assert.cc | 62 auto& return_op = block.back(); in ProcessFunction() local
65 assert_op->moveBefore(&return_op); in ProcessFunction()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | tpu_add_resource_device_attribute.cc | 70 auto return_op = llvm::dyn_cast<mlir::tf_device::ReturnOp>(use.getOwner()); in IdentifyConnectedAssignVariableOp() local
71 if (!return_op) continue; in IdentifyConnectedAssignVariableOp()
|
| D | merge_clusters.cc | 289 auto return_op = builder.create<mlir::tf_device::ReturnOp>( in CloneEmptyIfWithPredicate() local
293 builder.setInsertionPoint(return_op); in CloneEmptyIfWithPredicate()
543 auto return_op = in MergeClusters() local
545 if (!return_op) continue; in MergeClusters()
|
| D | layout_parsing.h | 87 mlir::func::ReturnOp return_op, const int return_index);
|
| D | layout_parsing.cc | 250 mlir::func::ReturnOp return_op, const int return_index) { in ExtractLayoutFromFunctionReturnAttr() argument
254 auto function = return_op->getParentOfType<mlir::func::FuncOp>(); in ExtractLayoutFromFunctionReturnAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | insert_main_function.cc | 120 auto& return_op = function.getBody().getBlocks().front().back(); in CreateMainFunction() local
121 result_types.append(return_op.getOperandTypes().begin(), in CreateMainFunction()
122 return_op.getOperandTypes().end()); in CreateMainFunction()
|
| /external/grpc-grpc/src/ruby/pb/test/ |
| D | client.rb | 419 return_op: true)
482 op = @stub.streaming_input_call(reqs, return_op: true)
493 op = @stub.full_duplex_call(ppp.each_item, return_op: true)
601 op = @stub.unary_call(req, metadata: metadata, return_op: true)
626 return_op: true)
|
| /external/tensorflow/tensorflow/core/transforms/shape_inference/ |
| D | pass.cc | 304 Operation *return_op = func.getBody()->getTerminator(); in runOnOperation() local
309 Type return_arg_type = return_op->getOperand(res_num).getType(); in runOnOperation()
320 TFOp(return_op).getNonControlOperands().getTypes()))); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 699 Operation* return_op = body.getTerminator(); in EliminatePassThroughResults() local
700 int num_results = return_op->getNumOperands(); in EliminatePassThroughResults()
709 for (OpOperand& operand : return_op->getOpOperands()) { in EliminatePassThroughResults()
745 rewriter.setInsertionPoint(return_op); in EliminatePassThroughResults()
747 rewriter.replaceOpWithNewOp<tf_device::ReturnOp>(return_op, cluster_vals); in EliminatePassThroughResults()
|
| /external/grpc-grpc/src/ruby/spec/ |
| D | google_rpc_status_utils_spec.rb | 198 op = stub.an_rpc(EchoMsg.new, return_op: true)
271 op = stub.an_rpc(EchoMsg.new, return_op: true)
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | graphdef_export.cc | 307 auto return_op = cast<ReturnOp>(func.getBody()->getTerminator()); in ExportFunction() local
311 TFOp(return_op).getNonControlOperands())) { in ExportFunction()
325 llvm::zip(return_op.control_ret_attrs().getAsRange<DictionaryAttr>(), in ExportFunction()
326 TFOp(return_op).getControlOperands())) { in ExportFunction()
|
| D | functiondef_export.cc | 234 auto return_op = in ConvertGenericFunctionToFunctionDef() local
237 for (auto &indexed_result : llvm::enumerate(return_op->getOperands())) { in ConvertGenericFunctionToFunctionDef()
|