Searched refs:FunctionCallFrame (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 149 FunctionCallFrame call_frame({DT_FLOAT, DT_FLOAT}, {DT_FLOAT}); in TEST_F()
176 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
225 FunctionCallFrame call_frame({DT_FLOAT}, {DT_FLOAT}); in TEST_F()
242 FunctionCallFrame call_frame({}, {}); in TEST_F()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 275 class FunctionCallFrame : public CallFrameInterface {
277 FunctionCallFrame(DataTypeSlice arg_types, DataTypeSlice ret_types);
278 ~FunctionCallFrame();
305 TF_DISALLOW_COPY_AND_ASSIGN(FunctionCallFrame);
|
| D | function.cc | 932 FunctionCallFrame::FunctionCallFrame(DataTypeSlice arg_types, in FunctionCallFrame() function in tensorflow::FunctionCallFrame
940 FunctionCallFrame::~FunctionCallFrame() {} in ~FunctionCallFrame()
942 Status FunctionCallFrame::SetArgs(gtl::ArraySlice<Tensor> args) { in SetArgs()
960 Status FunctionCallFrame::GetRetvals(std::vector<Tensor>* rets) const { in GetRetvals()
974 Status FunctionCallFrame::ConsumeRetvals(std::vector<Tensor>* rets, in ConsumeRetvals()
990 Status FunctionCallFrame::GetArg(int index, Tensor* val) const { in GetArg()
999 Status FunctionCallFrame::SetRetval(int index, const Tensor& val) { in SetRetval()
|
| D | function_test.cc | 907 TEST(FunctionCallFrame, Void_Void) { in TEST() argument
908 FunctionCallFrame frame({}, {}); in TEST()
920 TEST(FunctionCallFrame, Float_Float_Float) { in TEST() argument
921 FunctionCallFrame frame({DT_FLOAT, DT_FLOAT}, {DT_FLOAT}); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 971 FunctionCallFrame* frame = in RunRemote()
972 new FunctionCallFrame(fbody->arg_types, fbody->ret_types); in RunRemote()
1088 FunctionCallFrame* frame = in Run()
1089 new FunctionCallFrame(fbody->arg_types, fbody->ret_types); in Run()
|
| D | function_test.cc | 105 FunctionCallFrame frame(arg_types_, ret_types_); in Run()
301 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in InstantiateAndRunViaCallFrameInterface()
|
| D | direct_session.cc | 761 FunctionCallFrame call_frame(executors_and_keys->input_types, in Run()
|