Home
last modified time | relevance | path

Searched refs:GetArg (Results 1 – 17 of 17) sorted by relevance

/external/angle/src/compiler/translator/TranslatorMetalDirect/
DAddExplicitTypeCasts.cpp36 TIntermTyped &arg = GetArg(callNode, 0); in visitAggregatePost()
48 TIntermTyped &arg = GetArg(callNode, 0); in visitAggregatePost()
57 TIntermTyped &arg = GetArg(callNode, i); in visitAggregatePost()
65 TIntermTyped &arg = GetArg(callNode, 0); in visitAggregatePost()
DAstHelpers.h72 TIntermTyped &GetArg(const TIntermAggregate &call, size_t index);
DAstHelpers.cpp178 TIntermTyped &sh::GetArg(const TIntermAggregate &call, size_t index) in GetArg() function in sh
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc73 llvm::Optional<int> GetArg(int result_index) const { in GetArg() function in mlir::TF::detail::BacktrackAnalysisInfo
203 Optional<int> passthrough_arg = callee_info.getValue()->GetArg(res_index); in BacktrackValue()
340 auto passthrough_arg = func_info.GetArg(result.getResultNumber()); in ResourceAliasAnalysisInfo()
408 passthrough_args[result_index] = body_info.GetArg(result_index); in AnalyzeWhileLoop()
456 passthrough_args.emplace_back(info->GetArg(result.getResultNumber())); in AnalyzeFunctionalCaseOrIfOp()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Darg_op.cc45 OP_REQUIRES_OK(ctx, frame->GetArg(index_, &val)); in Compile()
/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc916 HasError(frame.GetArg(0, &v), "Invalid argument"); in TEST()
934 HasError(frame.GetArg(-1, &v), "Invalid argument"); in TEST()
935 HasError(frame.GetArg(2, &v), "Invalid argument"); in TEST()
936 TF_EXPECT_OK(frame.GetArg(0, &v)); in TEST()
938 TF_EXPECT_OK(frame.GetArg(1, &v)); in TEST()
Dfunction.h278 virtual Status GetArg(int index, const Tensor** val) = 0;
317 Status GetArg(int index, const Tensor** val) override;
Dfunction.cc1146 Status FunctionCallFrame::GetArg(int index, const Tensor** val) { in GetArg() function in tensorflow::FunctionCallFrame
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc161 TF_ASSERT_OK(call_frame.GetArg(0, &arg_0)); in TEST_F()
164 TF_ASSERT_OK(call_frame.GetArg(1, &arg_1)); in TEST_F()
Dmap_defun_op.cc93 Status GetArg(int index, const Tensor** val) override { in GetArg() function in tensorflow::data::MapDefunOp::MapFunctionCallFrame
Dsingle_threaded_executor.cc351 TF_CHECK_OK(args.call_frame->GetArg(i, &arg)); in Run()
Dcaptured_function.cc338 Status GetArg(int index, const Tensor** val) override { in GetArg() function in tensorflow::data::__anon5a9b84710111::OwnedArgsCallFrame
382 Status GetArg(int index, const Tensor** val) override { in GetArg() function in tensorflow::data::__anon5a9b84710111::BorrowedArgsCallFrame
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc65 OP_REQUIRES_OK(ctx, frame->GetArg(index_, &val)); in Compute()
Dfunctional_ops.cc457 Status GetArg(int index, const Tensor** val) override { in GetArg() function in tensorflow::__anonf02e412f0111::WhileOp::BodyFuncCallFrame
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc1575 Status s = frame->GetArg(i, &arg); in Run()
Ddirect_session.cc1878 Status GetArg(int index, const Tensor** val) override { in GetArg() function in tensorflow::DirectSession::RunCallableCallFrame
Dfunction_test.cc448 Status GetArg(int index, const Tensor** val) override { in GetArg() function in tensorflow::__anon196ffbc80111::ConsumeArgumentCallFrame