Home
last modified time | relevance | path

Searched refs:call_op (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_ophint_func_op.cc85 CallOp call_op, in BuildUnidirectionalSequenceRnnOp() argument
90 if (call_op.getNumOperands() != 5) return failure(); in BuildUnidirectionalSequenceRnnOp()
92 if (call_op.getNumResults() != 1) return failure(); in BuildUnidirectionalSequenceRnnOp()
95 Value input = call_op.getOperand(0); in BuildUnidirectionalSequenceRnnOp()
97 Value weight = call_op.getOperand(1); in BuildUnidirectionalSequenceRnnOp()
99 Value recurrent_weight = call_op.getOperand(2); in BuildUnidirectionalSequenceRnnOp()
101 Value bias = call_op.getOperand(3); in BuildUnidirectionalSequenceRnnOp()
103 Value hidden_state = call_op.getOperand(4); in BuildUnidirectionalSequenceRnnOp()
106 auto output_type = call_op.getResult(0).getType(); in BuildUnidirectionalSequenceRnnOp()
113 builder->setInsertionPoint(call_op.getOperation()); in BuildUnidirectionalSequenceRnnOp()
[all …]
/external/clang/test/SemaTemplate/
Dinstantiate-expr-1.cpp118 CallOperator call_op; in test_call_operator() local
119 int &ir = call_op(17); in test_call_operator()
123 void test_call_operator(CallOperator call_op, int i, double d) { in test_call_operator() argument
124 int &ir = test_call_operator<int&>(call_op, i); in test_call_operator()
125 double &dr = test_call_operator<double&>(call_op, d); in test_call_operator()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_simplifier.cc70 HloInstruction* call_op = create_call(0); in TryRemoveConditional() local
71 TF_RETURN_IF_ERROR(computation->ReplaceInstruction(conditional, call_op)); in TryRemoveConditional()
72 TF_RETURN_IF_ERROR(CallInliner::Inline(call_op).status()); in TryRemoveConditional()
86 HloInstruction* call_op = create_call(branch_index); in TryRemoveConditional() local
87 TF_RETURN_IF_ERROR(computation->ReplaceInstruction(conditional, call_op)); in TryRemoveConditional()
88 TF_RETURN_IF_ERROR(CallInliner::Inline(call_op).status()); in TryRemoveConditional()
Dwhile_loop_simplifier.cc496 auto call_op = computation->AddInstruction(HloInstruction::CreateCall( in TryRemoveWhileLoop() local
498 TF_RETURN_IF_ERROR(computation->ReplaceInstruction(while_op, call_op)); in TryRemoveWhileLoop()
500 CallInliner::Inline(call_op)); in TryRemoveWhileLoop()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_cfg.cc186 Operation* call_op = CallFn(loc, get_operand, then_fn, &builder); in LowerIfOp() local
188 auto get_then_result = [&](int i) { return call_op->getResult(i); }; in LowerIfOp()
193 call_op = CallFn(loc, get_operand, else_fn, &builder); in LowerIfOp()
195 auto get_else_result = [&](int i) { return call_op->getResult(i); }; in LowerIfOp()
Dshape_inference.cc448 } else if (auto call_op = dyn_cast<TF::PartitionedCallOp>(op)) { in PropagateShapeIntoAttachedFunctions() local
449 return PropagateShapeToFunctions(module, call_op.getOperandTypes(), in PropagateShapeIntoAttachedFunctions()
450 {call_op.f()}, graph_version, in PropagateShapeIntoAttachedFunctions()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc426 mlir::CallOp* call_op, xla::XlaBuilder* builder,
845 if (auto call_op = dyn_cast<mlir::CallOp>(inst)) { in Lower() local
846 return LowerFunctionCall(&call_op, builder, &value_map); in Lower()
891 mlir::CallOp* call_op, xla::XlaBuilder* builder, in LowerFunctionCall() argument
894 mlir::FuncOp callee = module_.lookupSymbol<mlir::FuncOp>(call_op->callee()); in LowerFunctionCall()
897 for (auto operand : call_op->getOperands()) { in LowerFunctionCall()
909 unsigned num_results = call_op->getNumResults(); in LowerFunctionCall()
912 value_map[call_op->getResult(i)] = xla::GetTupleElement(call_result, i); in LowerFunctionCall()
915 value_map[call_op->getResult(0)] = call_result; in LowerFunctionCall()
/external/grpc-grpc/src/ruby/pb/test/
Dclient.rb418 call_op = @stub.streaming_input_call(requests,
420 requests.call_op = call_op
421 resp = call_op.execute
/external/tensorflow/tensorflow/python/eager/
Dfunction.py840 call_op = outputs[0].op
841 return self._rewrite_forward_and_call_backward(call_op, *args)
/external/v8/src/compiler/
Djs-native-context-specialization.cc1582 const Operator* call_op = javascript()->Call( in ReduceJSGetIterator() local
1587 graph()->NewNode(call_op, load_property, receiver, n.feedback_vector(), in ReduceJSGetIterator()