- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GetTfGraphProducerVersion (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.h
// Returns TensorFlow GraphDef producer version for the given module. Returns an // error if the version information is missing for the module or is not valid. absl::StatusOr<int64_t> GetTfGraphProducerVersion(mlir::ModuleOp module); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
if (!bad_consumer_int_attr) return mlir::failure(); versions->mutable_bad_consumers()->Add(bad_consumer_int_attr.getInt()); } return mlir::success(); } absl::StatusOr<int64_t> GetTfGraphProducerVersion(mlir::ModuleOp module) { auto versions = module->getAttrOfType<::mlir::DictionaryAttr>("tf.versions"); if (!versions) { return errors::Internal( "Missing 'tf.versions' attribute on the module, abort.\n");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
<< "failed to create XlaContext resource: " << status.ToString(); } params_.step_container = step_container_.get(); absl::StatusOr<int64_t> version_or = tensorflow::GetTfGraphProducerVersion( op_->getParentOfType<mlir::ModuleOp>()); if (!version_or.ok()) { return emitError(op_->getLoc()) << version_or.status().ToString(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
mlir::mhlo::createSinkConstantsToControlFlowPass()); } Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, mlir::ModuleOp module) { auto producer_or = GetTfGraphProducerVersion(module); if (!producer_or.ok()) return producer_or.status(); int64_t producer_version = producer_or.value(); llvm::SmallVector<int64_t, 16> shape_backing;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)