Searched refs:gather_ops (Results 1 – 3 of 3) sorted by relevance
116 std::vector<GatherOperator*> gather_ops; in Run() local127 gather_ops.push_back(static_cast<GatherOperator*>(op)); in Run()132 for (auto* gather_op : gather_ops) { in Run()178 for (const auto& gather_op : gather_ops) { in Run()182 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_unpartitioned")); in Run()192 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted/perm")); in Run()194 AvailableArrayName(*model, gather_ops[0]->inputs[0] + "_permuted")); in Run()198 const auto& partition_array = model->GetArray(gather_ops[0]->inputs[0]); in Run()238 for (auto* gather_op : gather_ops) { in Run()
118 static const absl::flat_hash_set<string>* gather_ops = in SafeToConvertToNoOp()123 gather_ops->find(node.op()) != gather_ops->end(); in SafeToConvertToNoOp()
653 std::vector<std::string> gather_ops = {"Gather", "GatherNd", "GatherV2"}; in TEST_F() local655 for (const auto& op : gather_ops) { in TEST_F()