Home
last modified time | relevance | path

Searched refs:index_attr (Results 1 – 7 of 7) sorted by relevance

/external/llvm-project/mlir/test/IR/
Dattribute.mlir11 // CHECK-SAME: index_attr = 8 : index
12 index_attr = 8 : index,
22 index_attr = 8 : index,
30 index_attr = 8 : index,
129 index_attr = 8 : index,
142 index_attr = 8 : index,
Dcore-ops.mlir919 // CHECK: index_attr = 8 : index
920 } { index_attr = 8 : index }
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc234 AttrValue index_attr; in GetArgumentNode() local
235 index_attr.set_i(index); in GetArgumentNode()
236 (*node_def->mutable_attr())["index"] = index_attr; in GetArgumentNode()
269 AttrValue index_attr; in GetReturnNode() local
270 index_attr.set_i(index); in GetReturnNode()
271 (*node_def->mutable_attr())["index"] = index_attr; in GetReturnNode()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc270 DenseIntElementsAttr index_attr; in matchAndRewrite() local
271 if (!matchPattern(index, m_Constant(&index_attr))) return failure(); in matchAndRewrite()
272 indices.push_back(index_attr); in matchAndRewrite()
295 DenseIntElementsAttr index_attr = std::get<0>(it); in matchAndRewrite() local
306 for (auto index_item : llvm::zip(index_attr, items.getResults())) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc464 for (auto index_attr : tflite_indices_attr.getValue()) { in matchAndRewrite() local
465 IntegerAttr index = index_attr.cast<IntegerAttr>(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1959 DenseIntElementsAttr index_attr; in Verify() local
1960 if (matchPattern(index, m_Constant(&index_attr))) { in Verify()
1961 for (int32_t index : index_attr.getValues<int32_t>()) { in Verify()
/external/llvm-project/mlir/test/lib/Dialect/Test/
DTestOps.td231 IndexAttr:$index_attr,