Searched refs:TF_WhileParams (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/c/ |
D | c_api.h | 909 typedef struct TF_WhileParams { struct 929 } TF_WhileParams; argument 951 TF_CAPI_EXPORT extern TF_WhileParams TF_NewWhile(TF_Graph* g, TF_Output* inputs, 963 TF_CAPI_EXPORT extern void TF_FinishWhile(const TF_WhileParams* params, 970 TF_CAPI_EXPORT extern void TF_AbortWhile(const TF_WhileParams* params);
|
D | while_loop_test.cc | 48 params_.reset(new TF_WhileParams( in Init() 129 std::unique_ptr<TF_WhileParams> params_; 236 TF_WhileParams inner_params = in TEST_F()
|
D | c_api.cc | 1880 bool ValidateConstWhileParams(const TF_WhileParams& params, TF_Status* s) { in ValidateConstWhileParams() 1894 bool ValidateInputWhileParams(const TF_WhileParams& params, TF_Status* s) { in ValidateInputWhileParams() 1915 void FreeWhileResources(const TF_WhileParams* params) { in FreeWhileResources() 1923 TF_WhileParams EmptyWhileParams() { in EmptyWhileParams() 1930 TF_WhileParams TF_NewWhile(TF_Graph* g, TF_Output* inputs, int ninputs, in TF_NewWhile() 1971 TF_WhileParams params = {ninputs, cond_graph, cond_inputs, cond_output, in TF_NewWhile() 1986 void TF_FinishWhileHelper(const TF_WhileParams* params, TF_Status* status, in TF_FinishWhileHelper() 2059 void TF_FinishWhile(const TF_WhileParams* params, TF_Status* status, in TF_FinishWhile() 2074 void TF_AbortWhile(const TF_WhileParams* params) { FreeWhileResources(params); } in TF_AbortWhile()
|
D | c_api_function_test.cc | 913 std::unique_ptr<TF_WhileParams> params(new TF_WhileParams( in TEST_F()
|
/external/tensorflow/tensorflow/java/src/main/native/ |
D | graph_jni.cc | 272 TF_WhileParams params = TF_NewWhile(g, inputs.get(), ninputs, status); in Java_org_tensorflow_Graph_whileLoop()
|