- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TypeAttr (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
if (tensorflow::IsRefType(dtype)) dtype = tensorflow::RemoveRefType(dtype); value->set_type(dtype); return absl::OkStatus(); } Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type, AttrValue* value) { return ConvertAttribute(type.getValue(), remove_ref_type, value); } Status ConvertAttribute(const mlir::UnitAttr& attr, AttrValue* value) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
res_types, res_attrs))) return failure(); // Update the function type which has excluded the control args. func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType( arg_types, res_types))); // Update arg/result attributes. func.setAllArgAttrs(arg_attrs); func.setAllResultAttrs(res_attrs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
return builder.getI32IntegerAttr(value); } static mlir::Attribute BuildTypeAttr(tflite::TensorType value, mlir::Builder builder) { return mlir::TypeAttr::get(ConvertElementType(value, builder)); } static mlir::Attribute BuildTFL_AFAttr(tflite::ActivationFunctionType value, mlir::Builder builder) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0)