Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dif.cc34 int then_subgraph_index; member
41 op_data->then_subgraph_index = params->then_subgraph_index; in Init()
72 TF_LITE_ENSURE(context, op_data->then_subgraph_index < subgraphs->size()); in Prepare()
75 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
151 cond_value ? op_data->then_subgraph_index : op_data->else_subgraph_index; in Eval()
Dsubgraph_test_util.cc211 params->then_subgraph_index = 1; in BuildIfSubgraph()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h473 int then_subgraph_index; member
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dif_op.mlir63 // CHECK-NEXT: then_subgraph_index: 1,
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc740 params->then_subgraph_index = if_params->then_subgraph_index(); in ParseOpDataTfLite()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc854 int then_subgraph_index = subgraph_index_map_.at(op.then_branch().str()); in BuildIfOperator() local
856 auto builtin_options = tflite::CreateIfOptions(builder_, then_subgraph_index, in BuildIfOperator()
Dflatbuffer_import.cc576 uint32_t then_idx = opts->then_subgraph_index; in ConvertSubgraphIdxsToFunctionAttrs()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dtest_schema.fbs943 then_subgraph_index:int;
/external/tensorflow/tensorflow/lite/schema/
Dschema_v3a.fbs952 then_subgraph_index:int;
Dschema.fbs980 then_subgraph_index:int;
Dschema_generated.h9174 int32_t then_subgraph_index;
9177 : then_subgraph_index(0),
9188 int32_t then_subgraph_index() const {
9208 void add_then_subgraph_index(int32_t then_subgraph_index) {
9209 fbb_.AddElement<int32_t>(IfOptions::VT_THEN_SUBGRAPH_INDEX, then_subgraph_index, 0);
9228 int32_t then_subgraph_index = 0,
9232 builder_.add_then_subgraph_index(then_subgraph_index);
14257 { auto _e = then_subgraph_index(); _o->then_subgraph_index = _e; }
14269 auto _then_subgraph_index = _o->then_subgraph_index;