- Sort Score
- Result 10 results
- Languages All
Results 21 - 27 of 27 for NodeBuilder (0.13 sec)
-
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
{ GraphDefBuilder builder(GraphDefBuilder::kFailImmediately); Node* a = ops::SourceOp("UncompilableNullary", builder.opts().WithName("A")); // Builds a Symbolic gradient for Supported NodeBuilder b_builder("B", "SymbolicGradient", builder.opts().op_registry()); NameAttrList b_name_attr; b_name_attr.set_name("Supported"); b_builder.Attr("f", b_name_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// Don't set any value field in the proto, since it is only going to be used // for shape inference. GraphDefBuilder::Options options(graph_out, /*status=*/nullptr); NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", options.op_registry()); node_builder.Attr("dtype", data_type).Attr("value", dummy_proto);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
#include "tensorflow/core/platform/strcat.h" #include "tensorflow/core/protobuf/config.pb.h" #include "tensorflow/core/protobuf/tensorflow_server.pb.h" using tensorflow::FunctionDef; using tensorflow::Node; using tensorflow::NodeBuilder; using tensorflow::Status; using tensorflow::errors::InvalidArgument; namespace { typedef std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)> UniqueFuncPtr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
inputs.push_back(op.input(i)); } auto shapes = ShapeN(scope, inputs); const auto unique_name = scope.GetUniqueNameForOp("ConcatOffset"); auto builder = ::tensorflow::NodeBuilder(unique_name, "ConcatOffset") .Input(::tensorflow::ops::AsNodeOut(scope, op.input(dim_index))) .Input(::tensorflow::ops::AsNodeOutList(scope, shapes.output)); scope.UpdateBuilder(&builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis_test.cc
if (!scope.ok()) return; auto _output_index = ops::AsNodeOut(scope, output_index); if (!scope.ok()) return; Node* ret; const auto unique_name = scope.GetUniqueNameForOp("_SwitchN"); auto builder = NodeBuilder(unique_name, "_SwitchN") .Input(_data) .Input(_output_index) .Attr("num_outs", num_outs); scope.UpdateBuilder(&builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 51.6K bytes - Viewed (0) -
tensorflow/c/c_api.cc
using tensorflow::GraphDef; using tensorflow::mutex_lock; using tensorflow::NameRangeMap; using tensorflow::NameRangesForNode; using tensorflow::NewSession; using tensorflow::Node; using tensorflow::NodeBuilder; using tensorflow::OpDef; using tensorflow::PartialTensorShape; using tensorflow::RunMetadata; using tensorflow::RunOptions; using tensorflow::Session; using tensorflow::Status; using tensorflow::string;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
std::string new_node_name = update_inplace ? node->name() : GetUniqueNodeName(node->name(), index, node_name_map); Node* placeholder_node; NodeBuilder builder(new_node_name, "Placeholder"); builder.Attr("shape", shape); builder.Attr("dtype", dtype); TF_RETURN_IF_ERROR(builder.Finalize(graph_.get(), &placeholder_node));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)