- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 78 for nodedef (0.14 sec)
-
tensorflow/cc/tools/freeze_saved_model.cc
(*variable_name_to_value_map)[variable_names[i]] = outputs[i]; } return OkStatus(); } // Converts a Variable NodeDef into a Constant NodeDef. void ConvertVariableToConstant(const NodeDef& variable_node, const Tensor& variable_value, NodeDef* const_node) { const_node->set_name(variable_node.name()); const_node->set_op("Const");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
NodeDef* node_def) { const NameAttrList* name_attr; TF_RETURN_IF_ERROR(GetNodeAttr(node.attrs(), attr_name, &name_attr)); node_def->set_op(name_attr->name()); *(node_def->mutable_attr()) = name_attr->attr(); return absl::OkStatus(); } absl::StatusOr<std::vector<NodeDef>> MakeCallNodesFromAttribute( const Node& node, absl::string_view attr_name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
if (stateless && stateless.getValue()) *node_def->mutable_op() = "Stateless" + node_def->op(); } // Add inputs to the NodeDef based on the number of operands. This is required // as later when edges are added to the Node using Graph::AddEdge the // associated NodeDef is not updated. for (int i = 0, e = inst->getNumOperands(); i < e; ++i) { node_def->add_input(); }
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/lite/transforms/lift_tflite_flex_ops.cc
StringRef custom_options, Location loc, std::string& op_name, SmallVectorImpl<NamedAttribute>& attributes, tensorflow::NodeDef& node_def) { // The flexbuffer contains a vector where the first elements is the // op name and the second is a serialized NodeDef. const uint8_t* const opt_data = reinterpret_cast<const uint8_t*>(custom_options.data()); const size_t opt_size = custom_options.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
} } } return found_dtype && found_value; } bool IsAddN(const tensorflow::NodeDef& node_def, int n) { if (node_def.op() != "AddN" || node_def.name() != "add" || node_def.input_size() != n) { return false; } bool found_t = false; bool found_n = false; for (const auto& attr : node_def.attr()) { if (attr.first == "T") { if (attr.second.type() == tensorflow::DT_INT32) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 03:16:52 UTC 2021 - 17.8K bytes - Viewed (0) -
tensorflow/c/ops_test.cc
PartialTensorShape S(std::initializer_list<int64_t> dims) { return PartialTensorShape(dims); } PartialTensorShape Unknown() { return PartialTensorShape(); } TEST(OpsTest, ShapeInferenceWithRank) { NodeDef def; shape_inference::InferenceContext c(0, def, MakeOpDef(1, 0), {S({10, 20, 30})}, {}, {}, {}); shape_inference::ShapeHandle in0 = c.input(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
node_def->mutable_attr())); return node_def; } absl::StatusOr<std::unique_ptr<NodeDef>> Exporter::GetReturnNode( FuncOp function, Value operand, unsigned index, llvm::StringRef name) { auto node_def = std::make_unique<NodeDef>(); if (!name.empty()) node_def->set_name(std::string(ParseTensorName(name.str()).node())); else
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
node_def->mutable_attr())); return node_def; } absl::StatusOr<std::unique_ptr<NodeDef>> Exporter::GetReturnNode( FuncOp function, Value operand, unsigned index, llvm::StringRef name) { auto node_def = std::make_unique<NodeDef>(); if (!name.empty()) node_def->set_name(std::string(ParseTensorName(name.str()).node())); else
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc
namespace mlir::quant::stablehlo { namespace { using ::tensorflow::AssetFileDef; using ::tensorflow::GraphDef; using ::tensorflow::NodeDef; using ::tensorflow::SaverDef; using ::tensorflow::quantization::ExportedModel; using ::testing::HasSubstr; using ::testing::IsEmpty; using ::testing::SizeIs; using ::testing::StrEq; using ::tsl::protobuf::TextFormat; using ::tsl::testing::IsOk; using ::tsl::testing::StatusIs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
absl::StatusOr<std::unique_ptr<tensorflow::NodeDef>> node_def = tensorflow::ConvertTFDialectOpToNodeDef( op, /*name=*/"", /*ignore_unregistered_attrs=*/true); if (!node_def.ok()) { op->emitError("Failed to obtain TensorFlow NodeDef: " + node_def.status().ToString()); return false; } std::string node_def_str; if (!(*node_def)->SerializeToString(&node_def_str)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0)