Searched defs:gather_op (Results 1 – 4 of 4) sorted by relevance
132 for (auto* gather_op : gather_ops) { in Run() local178 for (const auto& gather_op : gather_ops) { in Run() local238 for (auto* gather_op : gather_ops) { in Run() local
31 auto* gather_op = model->operators[op_index].get(); in Run() local
1618 mhlo::GatherOp gather_op, ArrayRef<Value> args, in matchAndRewrite()
1233 tensorflow::NodeDef* gather_op = tensorflow_graph->add_node(); in ConvertGatherOperator() local