Searched refs:body_subgraph_index (Results 1 – 12 of 12) sorted by relevance
111 int body_subgraph_index; member120 op_data->body_subgraph_index = params->body_subgraph_index; in Init()140 TF_LITE_ENSURE(context, op_data->body_subgraph_index < subgraphs->size()); in Prepare()143 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Prepare()219 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Eval()
380 params->body_subgraph_index = 2; in BuildWhileSubgraph()568 params->body_subgraph_index = 2; in BuildWhileSubgraphWithDynamicTensor()
479 int body_subgraph_index; member
887 int body_subgraph_index = subgraph_index_map_.at(op.body().str()); in BuildWhileOperator() local889 builder_, cond_subgraph_index, body_subgraph_index) in BuildWhileOperator()908 auto body_subgraph_index = get_call_index(op.body().front()); in BuildWhileOperator() local910 if (!body_subgraph_index || !cond_subgraph_index) in BuildWhileOperator()915 *body_subgraph_index) in BuildWhileOperator()
589 uint32_t body_idx = opts->body_subgraph_index; in ConvertSubgraphIdxsToFunctionAttrs()
60 // CHECK-NEXT: body_subgraph_index: 2
751 params->body_subgraph_index = while_params->body_subgraph_index(); in ParseOpDataTfLite()
958 body_subgraph_index:int;
990 body_subgraph_index:int;
9295 int32_t body_subgraph_index;9298 body_subgraph_index(0) {9311 int32_t body_subgraph_index() const {9331 void add_body_subgraph_index(int32_t body_subgraph_index) {9332 fbb_.AddElement<int32_t>(WhileOptions::VT_BODY_SUBGRAPH_INDEX, body_subgraph_index, 0);9349 int32_t body_subgraph_index = 0) {9351 builder_.add_body_subgraph_index(body_subgraph_index);14313 { auto _e = body_subgraph_index(); _o->body_subgraph_index = _e; }14325 auto _body_subgraph_index = _o->body_subgraph_index;
949 body_subgraph_index:int;