Searched refs:argmin_op (Results 1 – 2 of 2) sorted by relevance
125 auto* argmin_op = static_cast<ArgMinOperator*>(op); in Run() local126 model->GetArray(op->outputs[0]).data_type = argmin_op->output_data_type; in Run()
1278 tensorflow::NodeDef* argmin_op = tensorflow_graph->add_node(); in ConvertArgMinOperator() local1279 argmin_op->set_op("ArgMin"); in ConvertArgMinOperator()1280 argmin_op->set_name(src_op.outputs[0]); in ConvertArgMinOperator()1282 *argmin_op->add_input() = src_op.inputs[0]; in ConvertArgMinOperator()1283 *argmin_op->add_input() = src_op.inputs[1]; in ConvertArgMinOperator()1284 (*argmin_op->mutable_attr())["T"].set_type( in ConvertArgMinOperator()1286 (*argmin_op->mutable_attr())["Tidx"].set_type( in ConvertArgMinOperator()1288 (*argmin_op->mutable_attr())["output_type"].set_type( in ConvertArgMinOperator()