Home
last modified time | relevance | path

Searched refs:float_attr (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Drewrite_util.h55 DenseFPElementsAttr float_attr; in IsConstantValueOf() local
56 if (matchPattern(value, m_Constant(&float_attr)) && float_attr.isSplat() && in IsConstantValueOf()
57 float_attr.getSplatValue<APFloat>().isExactlyValue(raw_value)) in IsConstantValueOf()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dnms_utils.cc152 auto float_attr = attrs.get(attribute).dyn_cast_or_null<FloatAttr>(); in AddFloatAttr() local
153 if (!float_attr) { in AddFloatAttr()
157 builder->Float(attribute.c_str(), float_attr.getValue().convertToFloat()); in AddFloatAttr()
173 auto float_attr = attrs.get(attribute).dyn_cast_or_null<FloatAttr>(); in HasFloatAttr() local
174 if (!float_attr) { in HasFloatAttr()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/
Dtflite_import_export.cc62 auto float_attr = in AttachCostPerDevice() local
64 device_costs.push_back({device_identifier, float_attr}); in AttachCostPerDevice()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc83 if (auto float_attr = attr.second.dyn_cast_or_null<FloatAttr>()) { in CreateTflFusableOpCustomOptions() local
84 fbb.Float(attr.first.data(), float_attr.getValue().convertToFloat()); in CreateTflFusableOpCustomOptions()