- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GetTensorFlowOpName (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
return mlir::isa<mlir::TF::TensorFlowRefType>( mlir::getElementTypeOrSelf(next_iter_sink.getInput().getType())); auto op_name_or_status = GetTensorFlowOpName(op->getName().getStringRef()); if (!op_name_or_status.ok()) return false; auto op_name = std::move(op_name_or_status).value(); if (op_name == "NextIteration")
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/export_utils.h
Status AddTensorFlowOpPrefix(std::string); // Maps an MLIR op name in the TensorFlow dialect or the TensorFlow control // dialect back into a TensorFlow valid op name. absl::StatusOr<llvm::StringRef> GetTensorFlowOpName(llvm::StringRef); // Converts an MLIR operation to TensorFlow NodeDef with given node name. This // name should be unique to the graph it is being inserted into.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
mlir::Operation* inst, llvm::StringRef name, bool ignore_unregistered_attrs) { TF_ASSIGN_OR_RETURN(auto node_def, GetOperationNodeDef(inst, name)); TF_ASSIGN_OR_RETURN(auto op_name, GetTensorFlowOpName(inst->getName().getStringRef())); const tensorflow::OpRegistrationData* op_reg_data = tensorflow::OpRegistry::Global()->LookUp(op_name.str()); TF_RETURN_IF_ERROR(GetAttrValuesFromOperation(inst, name, op_reg_data,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} else if (auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) { Operation& inner_op = island.GetBody().front(); auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef()); if (llvm::isa<FuncOp>(inner_op) && op_name.ok()) { // If it is TF Control dialect specific op, look up custom operation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} else if (auto island = llvm::dyn_cast<mlir::tf_executor::IslandOp>(inst)) { Operation& inner_op = island.GetBody().front(); auto op_name = GetTensorFlowOpName(inner_op.getName().getStringRef()); if (llvm::isa<FuncOp>(inner_op) && op_name.ok()) { // If it is TF Control dialect specific op, look up custom operation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)