Searched refs:SetRetval (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | retval_op.cc | 52 OP_REQUIRES_OK(ctx, frame->SetRetval(index_, input)); in Compile()
54 ctx->xla_context()->SetRetval(index_, ctx->InputExpression(0)); in Compile()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 914 HasError(frame.SetRetval(0, v), "Invalid argument"); in TEST()
939 HasError(frame.SetRetval(-1, v), "Invalid argument"); in TEST()
940 HasError(frame.SetRetval(1, v), "Invalid argument"); in TEST()
941 HasError(frame.SetRetval(0, test::AsTensor<int64>({-100})), in TEST()
946 TF_EXPECT_OK(frame.SetRetval(0, v)); in TEST()
947 HasError(frame.SetRetval(0, v), "has already been set"); in TEST()
|
| D | function.h | 266 virtual Status SetRetval(int index, const Tensor& val) = 0;
293 Status SetRetval(int index, const Tensor& val) override;
|
| D | function.cc | 999 Status FunctionCallFrame::SetRetval(int index, const Tensor& val) { in SetRetval() function in tensorflow::FunctionCallFrame
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.h | 65 void SetRetval(int index, const XlaExpression& expression);
|
| D | xla_context.cc | 66 void XlaContext::SetRetval(int index, const XlaExpression& expression) { in SetRetval() function in tensorflow::XlaContext
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 206 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::data::__anone81855e30111::MapDefunOp::MapFunctionCallFrame
|
| D | captured_function.cc | 186 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::data::__anoncde944130211::CallFrameBase
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 68 OP_REQUIRES_OK(ctx, frame->SetRetval(index_, val)); in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1769 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::DirectSession::RunCallableCallFrame
|