Searched refs:argmin_op (Results 1 – 2 of 2) sorted by relevance
120 auto* argmin_op = static_cast<ArgMinOperator*>(op); in Run() local121 model->GetArray(op->outputs[0]).data_type = argmin_op->output_data_type; in Run()
1265 tensorflow::NodeDef* argmin_op = tensorflow_graph->add_node(); in ConvertArgMinOperator() local1266 argmin_op->set_op("ArgMin"); in ConvertArgMinOperator()1267 argmin_op->set_name(src_op.outputs[0]); in ConvertArgMinOperator()1269 *argmin_op->add_input() = src_op.inputs[0]; in ConvertArgMinOperator()1270 *argmin_op->add_input() = src_op.inputs[1]; in ConvertArgMinOperator()1271 (*argmin_op->mutable_attr())["T"].set_type( in ConvertArgMinOperator()1273 (*argmin_op->mutable_attr())["Tidx"].set_type( in ConvertArgMinOperator()1275 (*argmin_op->mutable_attr())["output_type"].set_type( in ConvertArgMinOperator()