Searched refs:FunctionCallFrame (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 152 FunctionCallFrame call_frame({DT_FLOAT, DT_FLOAT}, {DT_FLOAT}); in TEST_F()
187 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
236 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
253 FunctionCallFrame call_frame({}, {}); in TEST_F()
268 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
|
| D | dataset_test_base.cc | 464 FunctionCallFrame frame(arg_types, ret_types); in RunFunction()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 300 class FunctionCallFrame : public CallFrameInterface {
302 FunctionCallFrame(DataTypeSlice arg_types, DataTypeSlice ret_types);
303 ~FunctionCallFrame() override;
330 TF_DISALLOW_COPY_AND_ASSIGN(FunctionCallFrame);
|
| D | function.cc | 1088 FunctionCallFrame::FunctionCallFrame(DataTypeSlice arg_types, in FunctionCallFrame() function in tensorflow::FunctionCallFrame
1096 FunctionCallFrame::~FunctionCallFrame() {} in ~FunctionCallFrame()
1098 Status FunctionCallFrame::SetArgs(gtl::ArraySlice<Tensor> args) { in SetArgs()
1116 Status FunctionCallFrame::GetRetvals(std::vector<Tensor>* rets) const { in GetRetvals()
1130 Status FunctionCallFrame::ConsumeRetvals(std::vector<Tensor>* rets, in ConsumeRetvals()
1146 Status FunctionCallFrame::GetArg(int index, const Tensor** val) { in GetArg()
1155 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 | 1019 FunctionCallFrame* frame = in RunRemote()
1020 new FunctionCallFrame(fbody->arg_types, fbody->ret_types); in RunRemote()
1136 FunctionCallFrame* frame = in Run()
1137 new FunctionCallFrame(fbody->arg_types, fbody->ret_types); in Run()
1282 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in RunSync()
|
| D | function_test.cc | 108 FunctionCallFrame frame(arg_types_, ret_types_); in Run()
308 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in InstantiateAndRunViaCallFrameInterface()
|
| D | direct_session.cc | 866 FunctionCallFrame call_frame(executors_and_keys->input_types, in Run()
|