Home
last modified time | relevance | path

Searched refs:else_subgraph_index (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dif.cc35 int else_subgraph_index; member
42 op_data->else_subgraph_index = params->else_subgraph_index; in Init()
73 TF_LITE_ENSURE(context, op_data->else_subgraph_index < subgraphs->size()); in Prepare()
76 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
151 cond_value ? op_data->then_subgraph_index : op_data->else_subgraph_index; in Eval()
Dsubgraph_test_util.cc212 params->else_subgraph_index = 2; in BuildIfSubgraph()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h474 int else_subgraph_index; member
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dif_op.mlir64 // CHECK-NEXT: else_subgraph_index: 2
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc741 params->else_subgraph_index = if_params->else_subgraph_index(); in ParseOpDataTfLite()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc855 int else_subgraph_index = subgraph_index_map_.at(op.else_branch().str()); in BuildIfOperator() local
857 else_subgraph_index) in BuildIfOperator()
Dflatbuffer_import.cc578 uint32_t else_idx = opts->else_subgraph_index; in ConvertSubgraphIdxsToFunctionAttrs()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v3a.fbs953 else_subgraph_index:int;
Dschema.fbs981 else_subgraph_index:int;
Dschema_generated.h9175 int32_t else_subgraph_index;
9178 else_subgraph_index(0) {
9191 int32_t else_subgraph_index() const {
9211 void add_else_subgraph_index(int32_t else_subgraph_index) {
9212 fbb_.AddElement<int32_t>(IfOptions::VT_ELSE_SUBGRAPH_INDEX, else_subgraph_index, 0);
9229 int32_t else_subgraph_index = 0) {
9231 builder_.add_else_subgraph_index(else_subgraph_index);
14258 { auto _e = else_subgraph_index(); _o->else_subgraph_index = _e; }
14270 auto _else_subgraph_index = _o->else_subgraph_index;
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dtest_schema.fbs944 else_subgraph_index:int;