Home
last modified time | relevance | path

Searched refs:tensor_proto_attr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dconvert_tf_quant_ops_to_mhlo.cc93 TF::TensorProtoAttr tensor_proto_attr; in matchAndRewrite() local
94 if (!matchPattern(this_op.rhs(), m_Constant(&tensor_proto_attr))) in matchAndRewrite()
129 llvm::StringRef mangled_tensor = tensor_proto_attr.getValue(); in matchAndRewrite()
H A Dquantize_composite_functions.cc715 Attribute tensor_proto_attr = Quantize(attr, tensor_qtype); in matchAndRewrite() local
716 if (!tensor_proto_attr) { in matchAndRewrite()
733 if (!mlir::tfg::ConvertToTensorProto(tensor_proto_attr, &tensor_proto) in matchAndRewrite()
740 tensor_proto_attr = ElementsAttr(TF::TensorProtoAttr::get( in matchAndRewrite()
744 rewriter.create<TF::ConstOp>(loc, new_type, tensor_proto_attr); in matchAndRewrite()