- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ConvertToTensorProto (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
value->set_f(attr.getValueAsDouble()); return absl::OkStatus(); } Status ConvertAttribute(const mlir::ElementsAttr& attr, AttrValue* value) { return ConvertToTensorProto(attr, value->mutable_tensor()); } Status ConvertAttribute(const mlir::TF::PlaceholderAttr& attr, AttrValue* value) { value->set_placeholder(attr.getValue().str());
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/utils/convert_tensor.cc
} else { output->reserve(attr.getNumElements()); for (const T value : attr.getValues<T>()) output->push_back(Eigen::numext::bit_cast<uint8_t>(value)); } } Status ConvertToTensorProto(const ElementsAttr attr, TensorProto* output) { auto type = attr.getShapedType(); auto shape = type.getShape(); DataType output_dtype; TF_RETURN_IF_ERROR(ConvertToDataType(type, &output_dtype));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0)