Home
last modified time | relevance | path

Searched defs:call_op (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsignature_def_function.cc41 tensorflow::ImmediateOpPtr call_op; in TF_SignatureDefFunctionMakeCallOp() local
Dconcrete_function.cc40 tensorflow::ImmediateOpPtr call_op; in TF_ConcreteFunctionMakeCallOp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_inline_tpu_island.cc57 auto walk_result = getOperation().walk([&](TF::PartitionedCallOp call_op) { in runOnOperation()
Dtpu_sharding_identification_pass.cc99 if (auto call_op = llvm::dyn_cast<CallOpInterface>(owner)) { in GetXlaShardingFromArg() local
230 if (auto call_op = llvm::dyn_cast_or_null<CallOpInterface>(def)) { in GetXlaShardingFromRetval() local
Dtpu_identity_pruning.cc54 src.walk([&reachable_funcs, &funcs_to_visit](CallOpInterface call_op) { in CollectReachableFunctions()
Dexecutor_tpuv1_outline_tpu_island.cc139 auto call_op = builder.create<mlir::TF::PartitionedCallOp>( in runOnOperation() local
Dresource_op_lifting.cc1166 CallOpType call_op, PartitionedCallLiftingInfo& lifting_info) { in UpdatePartitionedCallOpWithNewCallee()
1208 CallOpType call_op, FuncOp callee, ModuleOp module, bool vars_initialized, in HandlePartitionedCallOp()
1265 } else if (auto call_op = llvm::dyn_cast<TF::PartitionedCallOp>(&op)) { in HoistForControlFlow() local
1277 } else if (auto call_op = in HoistForControlFlow() local
Dshape_inference.cc682 bool ShapeInference::InferShapeForCall(CallOpInterface call_op) { in InferShapeForCall()
1210 void ShapeInference::PropagateConstantToCallee(CallOpInterface call_op, in PropagateConstantToCallee()
1238 void ShapeInference::PropagateConstantFromCallee(CallOpInterface call_op, in PropagateConstantFromCallee()
1351 } else if (auto call_op = dyn_cast<CallOpInterface>(op)) { in PropagateShapeIntoAttachedFunctions() local
Dfunctional_control_flow_to_cfg.cc161 Operation* call_op = CallFn(loc, get_operand, op.then_function(), &builder); in LowerIfOp() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc226 PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op, in CollectInputsAndAttributes()
359 PatternRewriter& rewriter, CallOp call_op, in CreateAndReplaceOp()
409 CallOp call_op, PatternRewriter& rewriter) const { in matchAndRewrite()
Ddecompose.cc251 auto walk_result = func.walk([&](CallOp call_op) { in InlineTFRFuncCalls()
/external/llvm-project/clang/test/SemaTemplate/
Dinstantiate-expr-1.cpp118 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/clang/test/SemaTemplate/
Dinstantiate-expr-1.cpp118 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/
Dconditional_simplifier.cc93 HloInstruction* call_op = create_call(0); in TryRemoveConditional() local
109 HloInstruction* call_op = create_call(branch_index); in TryRemoveConditional() local
Dwhile_loop_simplifier.cc702 auto call_op = computation->AddInstruction(HloInstruction::CreateCall( in TryRemoveWhileLoop() local
/external/grpc-grpc/src/ruby/pb/test/
Dclient.rb236 attr_accessor :call_op accessor in WriteFlagSettingStreamingInputEnumerable
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib_test.cc183 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> call_op( in TEST() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc161 Operation* call_op, int expected_num_outputs, in CheckOutputConsumer()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_supported_values_test.py106 def call_op(self, op, *args, **kwargs): member in WrappedTensorOpDispatcher
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1264 if (auto call_op = dyn_cast<mlir::CallOp>(inst)) { in Lower() local
1357 mlir::CallOp call_op, xla::XlaBuilder* builder, in LowerFunctionCall()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc1181 auto call_op = op_builder.create<mlir::CallOp>( in AddCallOpInWhileOpRegion() local
Dflatbuffer_export.cc904 if (auto call_op = dyn_cast<mlir::CallOp>(b.front())) in BuildWhileOperator() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc749 auto call_op = rewriter.create<PartitionedCallOp>( in matchAndRewrite() local
2499 auto call_op = rewriter.create<typename decltype(op_type)::CallOp>( in matchAndRewrite() local