Searched refs:index_attr (Results 1 – 4 of 4) sorted by relevance
80 AttrValue index_attr; in CreateConfig() local81 index_attr.set_i(index); in CreateConfig()82 (*custom_optimizer->mutable_parameter_map())[kIndex] = index_attr; in CreateConfig()
195 DenseIntElementsAttr index_attr; in matchAndRewrite() local196 if (!matchPattern(index, m_Constant(&index_attr))) return matchFailure(); in matchAndRewrite()197 indices.push_back(index_attr); in matchAndRewrite()219 DenseIntElementsAttr index_attr = std::get<0>(it); in matchAndRewrite() local229 for (auto index_item : llvm::zip(index_attr, items.getResults())) { in matchAndRewrite()
260 AttrValue index_attr; in GetArgumentNode() local261 index_attr.set_i(index); in GetArgumentNode()262 (*node_def->mutable_attr())["index"] = index_attr; in GetArgumentNode()297 AttrValue index_attr; in GetReturnNode() local298 index_attr.set_i(index); in GetReturnNode()299 (*node_def->mutable_attr())["index"] = index_attr; in GetReturnNode()
870 DenseIntElementsAttr index_attr; in Verify() local871 if (matchPattern(index, m_Constant(&index_attr))) { in Verify()872 for (int32_t index : index_attr.getValues<int32_t>()) { in Verify()