| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | xla_cluster_formation.cc | 31 void EncapsulatePartitionedCall(TF::StatefulPartitionedCallOp call_op) { in EncapsulatePartitionedCall()
53 module.walk([&](TF::StatefulPartitionedCallOp call_op) { in runOnOperation()
60 for (auto call_op : ops) { in runOnOperation() local
|
| D | convert_launch_func_to_tf_call.cc | 52 auto call_op = builder.create<TF::PartitionedCallOp>( in runOnOperation() local
|
| D | executor_tpuv1_inline_tpu_island.cc | 57 auto walk_result = getOperation().walk([&](TF::PartitionedCallOp call_op) { in runOnOperation()
|
| D | tpu_identity_pruning.cc | 57 src.walk([&reachable_funcs, &funcs_to_visit](CallOpInterface call_op) { in CollectReachableFunctions()
|
| D | tpu_sharding_identification_pass.cc | 160 if (auto call_op = llvm::dyn_cast<CallOpInterface>(owner)) { in GetXlaShardingFromArg() local
335 if (auto call_op = llvm::dyn_cast_or_null<CallOpInterface>(def)) { in GetXlaShardingFromRetval() local
|
| D | executor_tpuv1_outline_tpu_island.cc | 163 auto call_op = builder.create<mlir::TF::PartitionedCallOp>( in runOnOperation() local
|
| D | resource_op_lifting.cc | 1099 CallOpType call_op, PartitionedCallLiftingInfo& lifting_info) { in UpdatePartitionedCallOpWithNewCallee()
1143 CallOpType call_op, func::FuncOp callee, ModuleOp module, in HandlePartitionedCallOp()
1201 } else if (auto call_op = llvm::dyn_cast<TF::PartitionedCallOp>(&op)) { in HoistForControlFlow() local
1213 } else if (auto call_op = in HoistForControlFlow() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | cluster_function_conversion.cc | 111 mlir::Operation* call_op) { in MaybeSkipXlaCompilation()
137 auto call_op = builder->create<mlir::TF::StatefulPartitionedCallOp>( in ReplaceClusterWithPartitionCallOp() local
|
| D | op_utils.cc | 43 auto call_op = llvm::dyn_cast<mlir::CallOpInterface>(op); in MaybeFindFunction() local
|
| D | propagate_device_id_to_function_args.cc | 63 if (auto call_op = in FindFunctionsToRewrite() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | signature_def_function.cc | 41 tensorflow::ImmediateOpPtr call_op; in TF_SignatureDefFunctionMakeCallOp() local
|
| D | concrete_function.cc | 40 tensorflow::ImmediateOpPtr call_op; in TF_ConcreteFunctionMakeCallOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/ops/ |
| D | tf_op_quant_spec.cc | 31 if (auto call_op = dyn_cast<TF::PartitionedCallOp>(op)) { in GetTFOpQuantSpec() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 229 PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op, in CollectInputsAndAttributes()
375 PatternRewriter& rewriter, CallOp call_op, in CreateAndReplaceOp()
425 CallOp call_op, PatternRewriter& rewriter) const { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | quantize_composite_functions.cc | 255 bool IsQuantizedCallforDynamicRange(TF::PartitionedCallOp call_op) { in IsQuantizedCallforDynamicRange()
284 bool IsQuantizedCallforStaticRange(TF::PartitionedCallOp call_op) { in IsQuantizedCallforStaticRange()
420 LogicalResult matchAndRewrite(TF::PartitionedCallOp call_op, in matchAndRewrite()
609 TF::PartitionedCallOp call_op, const SmallVector<Value, 4>& args, in mergeDequantizeOpFollowingQuantizedFunction()
|
| D | lift_quantizable_spots_as_functions_drq.cc | 92 LogicalResult matchAndRewrite(TF::PartitionedCallOp call_op, in matchAndRewrite()
|
| D | insert_main_function.cc | 213 auto call_op = builder.create<TF::PartitionedCallOp>( in CreateMainFunction() local
|
| /external/clang/test/SemaTemplate/ |
| D | instantiate-expr-1.cpp | 118 CallOperator call_op; in test_call_operator() local
123 void test_call_operator(CallOperator call_op, int i, double d) { in test_call_operator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier.cc | 468 HloInstruction* call_op = create_call(0); in TryRemoveConditional() local
484 HloInstruction* call_op = create_call(branch_index); in TryRemoveConditional() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | remove_tf_if_const_args.cc | 152 auto call_op = builder.create<mlir::TF::StatefulPartitionedCallOp>( in CreateBranchFunction() local
|
| D | merge_tf_if_ops.cc | 230 auto call_op = builder.create<mlir::TF::PartitionedCallOp>( in CreateBranchFunction() local
|
| /external/tensorflow/tensorflow/cc/experimental/libtf/ |
| D | function.cc | 59 auto call_op = AbstractOperationPtr(ctx->CreateOperation()); in ExecuteFunction() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib_test.cc | 184 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> call_op( in TEST() local
|
| /external/grpc-grpc/src/ruby/pb/test/ |
| D | client.rb | 239 attr_accessor :call_op accessor in WriteFlagSettingStreamingInputEnumerable
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | lift_as_function_call_utils.cc | 56 TF::PartitionedCallOp call_op = builder.create<TF::PartitionedCallOp>( in createFusedFnCall() local
|