Zenn
Open3

TensorFlow Lite の IF OP 解析

TensorFlow
解析
TensorFlowLite
op
PINTOPINTO
  • INPUTS の１個目の入力値は boolean で condition
  • ノードの最初の入力は条件です。残りの入力は分岐サブグラフに渡される。したがって、サブグラフの入力数は、ノードの入力数-1となる。
PINTOPINTO
  Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_);
  auto* subgraphs = this_subgraph->GetSubgraphs();

  Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get();
  Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get();