Searched refs:onehot_op (Results 1 – 1 of 1) sorted by relevance
1445 tensorflow::NodeDef* onehot_op = tensorflow_graph->add_node(); in ConvertOneHotOperator() local1446 onehot_op->set_op("OneHot"); in ConvertOneHotOperator()1447 onehot_op->set_name(src_op.outputs[0]); in ConvertOneHotOperator()1450 *onehot_op->add_input() = input; in ConvertOneHotOperator()1452 (*onehot_op->mutable_attr())["T"].set_type( in ConvertOneHotOperator()1454 (*onehot_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertOneHotOperator()