- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 157 for node_def (0.14 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/graph_def.h
void MutateNodeDefs(tensorflow::GraphDef& graph_def, FuncT&& func) { for (tensorflow::NodeDef& node_def : *graph_def.mutable_node()) { func(node_def); } for (tensorflow::FunctionDef& function_def : *graph_def.mutable_library()->mutable_function()) { for (tensorflow::NodeDef& node_def : *function_def.mutable_node_def()) { func(node_def); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 28 18:38:06 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/graph_def_test.cc
MutateNodeDefs(graph_def, [](NodeDef& node_def) { node_def.set_name("bar"); }); ASSERT_THAT(graph_def.library().function(), SizeIs(1)); ASSERT_THAT(graph_def.library().function()[0].node_def(), SizeIs(1)); EXPECT_THAT(graph_def.library().function()[0].node_def()[0].name(), StrEq("bar")); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 28 18:38:06 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
} } } } } // Upgrade nodes in the GraphDef. for (auto& node_def : *gdef.mutable_node()) { const OpDef* op_def = nullptr; TF_RETURN_IF_ERROR(flib_def->LookUpOpDef(node_def.op(), &op_def)); // TODO(b/197144710): improve the shared_name attr, each op may use the // shared_name differently.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 14:33:47 UTC 2024 - 6.2K 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/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/jit/xla_kernel_creator.cc
return CanCreateXlaKernel(props->node_def) && !XlaOpRegistry::IsCompilationDevice(flr.device()->device_type()); } static Status CreateXlaKernel(FunctionLibraryRuntime* flr, const NodeDef& node_def, std::unique_ptr<OpKernel>* kernel) { if (!CanCreateXlaKernel(node_def)) { return errors::Internal("Invalid node: ", node_def.ShortDebugString()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/c_test_util.h
bool IsPlaceholder(const tensorflow::NodeDef& node_def); bool IsScalarConst(const tensorflow::NodeDef& node_def, int v); bool IsAddN(const tensorflow::NodeDef& node_def, int n); bool IsNeg(const tensorflow::NodeDef& node_def, const string& input); bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def); bool GetNodeDef(TF_Operation* oper, tensorflow::NodeDef* node_def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 09 01:06:53 UTC 2018 - 6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator.h
#include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/framework/node_properties.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow { class FunctionLibraryRuntime; class OpKernel; class XlaKernelCreator : public CustomKernelCreator { public: // Given a NodeDef 'node_def' and the function library runtime 'flr', returns
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 26 19:43:17 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
absl::StatusOr<FunctionDef> TFRDecomposeContext::ExpandNode( const NodeDef& node_def, StringPiece func_name) { const OpDef* op_def; TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef(node_def.op(), &op_def)); DataTypeVector input_dtys, output_dtys; TF_RETURN_IF_ERROR(InputTypesForNode(node_def, *op_def, &input_dtys)); TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, &output_dtys));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0)