Searched refs:onehot_op (Results 1 – 1 of 1) sorted by relevance
1415 tensorflow::NodeDef* onehot_op = tensorflow_graph->add_node(); in ConvertOneHotOperator() local1416 onehot_op->set_op("OneHot"); in ConvertOneHotOperator()1417 onehot_op->set_name(src_op.outputs[0]); in ConvertOneHotOperator()1420 *onehot_op->add_input() = input; in ConvertOneHotOperator()1422 (*onehot_op->mutable_attr())["T"].set_type( in ConvertOneHotOperator()1424 (*onehot_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertOneHotOperator()