Searched refs:GetArg (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | arg_op.cc | 43 OP_REQUIRES_OK(ctx, frame->GetArg(index_, &val)); in Compile()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 913 HasError(frame.GetArg(0, &v), "Invalid argument"); in TEST()
931 HasError(frame.GetArg(-1, &v), "Invalid argument"); in TEST()
932 HasError(frame.GetArg(2, &v), "Invalid argument"); in TEST()
933 TF_EXPECT_OK(frame.GetArg(0, &v)); in TEST()
935 TF_EXPECT_OK(frame.GetArg(1, &v)); in TEST()
|
| D | function.h | 265 virtual Status GetArg(int index, Tensor* val) const = 0;
292 Status GetArg(int index, Tensor* val) const override;
|
| D | function.cc | 990 Status FunctionCallFrame::GetArg(int index, Tensor* val) const { in GetArg() function in tensorflow::FunctionCallFrame
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 225 Status GetArg(int index, Tensor* val) const override { in GetArg() function in tensorflow::data::__anoncde944130211::OwnedArgsCallFrame
261 Status GetArg(int index, Tensor* val) const override { in GetArg() function in tensorflow::data::__anoncde944130211::BorrowedArgsCallFrame
|
| D | map_defun_op.cc | 180 Status GetArg(int index, Tensor* val) const override { in GetArg() function in tensorflow::data::__anone81855e30111::MapDefunOp::MapFunctionCallFrame
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 47 OP_REQUIRES_OK(ctx, frame->GetArg(index_, &val)); in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1757 Status GetArg(int index, Tensor* val) const override { in GetArg() function in tensorflow::DirectSession::RunCallableCallFrame
|