Searched refs:FunctionCallFrame (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 150 FunctionCallFrame call_frame({DT_FLOAT, DT_FLOAT}, {DT_FLOAT}); in TEST_F()
177 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
226 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
243 FunctionCallFrame call_frame({}, {}); in TEST_F()
|
| D | dataset_test_base.cc | 455 FunctionCallFrame frame(arg_types, ret_types); in RunFunction()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 286 class FunctionCallFrame : public CallFrameInterface {
288 FunctionCallFrame(DataTypeSlice arg_types, DataTypeSlice ret_types);
289 ~FunctionCallFrame() override;
316 TF_DISALLOW_COPY_AND_ASSIGN(FunctionCallFrame);
|
| D | function.cc | 1063 FunctionCallFrame::FunctionCallFrame(DataTypeSlice arg_types, in FunctionCallFrame() function in tensorflow::FunctionCallFrame
1071 FunctionCallFrame::~FunctionCallFrame() {} in ~FunctionCallFrame()
1073 Status FunctionCallFrame::SetArgs(gtl::ArraySlice<Tensor> args) { in SetArgs()
1091 Status FunctionCallFrame::GetRetvals(std::vector<Tensor>* rets) const { in GetRetvals()
1105 Status FunctionCallFrame::ConsumeRetvals(std::vector<Tensor>* rets, in ConsumeRetvals()
1121 Status FunctionCallFrame::GetArg(int index, Tensor* val) const { in GetArg()
1130 Status FunctionCallFrame::SetRetval(int index, const Tensor& val) { in SetRetval()
|
| D | function_test.cc | 910 TEST(FunctionCallFrame, Void_Void) { in TEST() argument
911 FunctionCallFrame frame({}, {}); in TEST()
923 TEST(FunctionCallFrame, Float_Float_Float) { in TEST() argument
924 FunctionCallFrame frame({DT_FLOAT, DT_FLOAT}, {DT_FLOAT}); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 1037 FunctionCallFrame* frame = in RunRemote()
1038 new FunctionCallFrame(fbody->arg_types, fbody->ret_types); in RunRemote()
1154 FunctionCallFrame* frame = in Run()
1155 new FunctionCallFrame(fbody->arg_types, fbody->ret_types); in Run()
|
| D | function_test.cc | 112 FunctionCallFrame frame(arg_types_, ret_types_); in Run()
306 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in InstantiateAndRunViaCallFrameInterface()
|
| D | direct_session.cc | 852 FunctionCallFrame call_frame(executors_and_keys->input_types, in Run()
|