| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | retval_op.cc | 54 OP_REQUIRES_OK(ctx, frame->SetRetval(index_, input)); in Compile()
56 ctx->xla_context()->SetRetval(index_, ctx->InputExpression(0)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.h | 75 void SetRetval(int index, const XlaExpression& expression);
|
| D | xla_context.cc | 72 void XlaContext::SetRetval(int index, const XlaExpression& expression) { in SetRetval() function in tensorflow::XlaContext
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 917 HasError(frame.SetRetval(0, *v), "Invalid argument"); in TEST()
942 HasError(frame.SetRetval(-1, w), "Invalid argument"); in TEST()
943 HasError(frame.SetRetval(1, w), "Invalid argument"); in TEST()
944 HasError(frame.SetRetval(0, test::AsTensor<int64>({-100})), in TEST()
949 TF_EXPECT_OK(frame.SetRetval(0, *v)); in TEST()
950 HasError(frame.SetRetval(0, *v), "has already been set"); in TEST()
|
| D | function.h | 291 virtual Status SetRetval(int index, const Tensor& val) = 0;
318 Status SetRetval(int index, const Tensor& val) override;
|
| D | function.cc | 1155 Status FunctionCallFrame::SetRetval(int index, const Tensor& val) { in SetRetval() function in tensorflow::FunctionCallFrame
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 122 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::data::MapDefunOp::MapFunctionCallFrame
|
| D | captured_function.cc | 298 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::data::__anon5a9b84710111::CallFrameBase
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 84 OP_REQUIRES_OK(ctx, frame->SetRetval(index_, val)); in Compute()
|
| D | functional_ops.cc | 475 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::__anonf02e412f0111::WhileOp::BodyFuncCallFrame
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 1603 Status s = frame->SetRetval(i, (*rets)[i]); in Run()
|
| D | direct_session.cc | 1887 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::DirectSession::RunCallableCallFrame
|
| D | function_test.cc | 456 Status SetRetval(int index, const Tensor& val) override { in SetRetval() function in tensorflow::__anon196ffbc80111::ConsumeArgumentCallFrame
|