Home
last modified time | relevance | path

Searched refs:else_subgraph_index (Results 1 – 12 of 12) 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.cc206 params->else_subgraph_index = 2; in BuildIfSubgraph()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h477 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.cc725 params->else_subgraph_index = if_params->else_subgraph_index(); in ParseOpDataTfLite()
1397 params->else_subgraph_index = schema_params->else_subgraph_index(); in ParseIf()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc908 int else_subgraph_index = subgraph_index_map_.at(op.else_branch().str()); in BuildIfOperator() local
910 else_subgraph_index) in BuildIfOperator()
Dflatbuffer_import.cc713 uint32_t else_idx = opts->else_subgraph_index; in ConvertSubgraphIdxsToFunctionAttrs()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v3a.fbs953 else_subgraph_index:int;
Dschema.fbs1007 else_subgraph_index:int;
Dschema_v3b.fbs1006 else_subgraph_index:int;
Dschema_generated.h9343 int32_t else_subgraph_index;
9346 else_subgraph_index(0) {
9359 int32_t else_subgraph_index() const {
9379 void add_else_subgraph_index(int32_t else_subgraph_index) {
9380 fbb_.AddElement<int32_t>(IfOptions::VT_ELSE_SUBGRAPH_INDEX, else_subgraph_index, 0);
9397 int32_t else_subgraph_index = 0) {
9399 builder_.add_else_subgraph_index(else_subgraph_index);
14833 { auto _e = else_subgraph_index(); _o->else_subgraph_index = _e; }
14845 auto _else_subgraph_index = _o->else_subgraph_index;
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dtest_schema.fbs943 else_subgraph_index:int;