Searched refs:int_attr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | raise-custom-ops.mlir | 8 …%2 = "tf.MyCustomOp"(%1, %0) {fused_activation_function = "RELU", int_attr = 2 : i32} : (tensor<4…
10 …"tf.MyCustomOp"(%1, %0) {fused_activation_function = "RELU", int_attr = 2 : i32} : (tensor<4xf32>…
16 …tf.MyCustomOp"(%[[MUL]], %[[CST]]) {fused_activation_function = "RELU", int_attr = 2 : i32} : (ten…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | nms_utils.cc | 137 auto int_attr = attrs.get(attribute).dyn_cast_or_null<IntegerAttr>(); in AddIntAttr() local
138 if (!int_attr) { in AddIntAttr()
142 builder->Int(attribute.c_str(), int_attr.getInt()); in AddIntAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | custom_op_with_tflite_op.mlir | 109 …%2 = "tf.MyCustomOp"(%1, %0) {fused_activation_function = "RELU", int_attr = 2 : i32} : (tensor<4…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2126 auto int_attr = entry.value().cast<IntegerAttr>(); in Verify() local
2127 if (int_attr.getInt() < min) in Verify()
2130 << min << ", but got " << int_attr.getInt() << " at index " in Verify()
|
| D | tf_ops_a_m.cc | 2059 auto int_attr = attr.cast<DenseIntElementsAttr>(); in fold() local
2061 for (const auto val : int_attr.getValues<int32_t>()) { in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 817 auto int_attr = input.cast<DenseIntElementsAttr>(); in SliceDenseIntElementsAttrColumn2D() local
818 auto shaped_type = int_attr.getType(); in SliceDenseIntElementsAttrColumn2D()
826 for (auto it : llvm::enumerate(int_attr.getIntValues())) { in SliceDenseIntElementsAttrColumn2D()
3532 auto int_attr = attr.cast<DenseIntElementsAttr>(); in ConvertAxisAttr() local
3536 axis.reserve(int_attr.getNumElements()); in ConvertAxisAttr()
3539 for (auto val : int_attr.getValues<APInt>()) { in ConvertAxisAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1837 IntegerAttr int_attr = *attr.getValues<IntegerAttr>().begin(); in ExtractConstantIntFromTensor() local
1838 return int_attr.getValue().getSExtValue(); in ExtractConstantIntFromTensor()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops_test.py | 947 op = test_ops.int_attr().op
|