- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for NodeBuilder (0.21 sec)
-
tensorflow/cc/ops/while_loop.cc
// not-yet-created next iteration nodes). Use the underlying NodeBuilder API // directly to create the back edge. NodeBuilder::NodeOut enter_input(enter_output.node(), enter_output.index()); const int next_output_index = 0; DataType dtype = enter_output.node()->output_type(0); NodeBuilder::NodeOut next_input(NextIterationName(scope, loop_var_idx),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_graph_test.cc
return errors::Unimplemented("Stream not seekable"); } private: string& str_; }; TEST(Dump, TextualIrToFileSuccess) { Graph graph(OpRegistry::Global()); Node* node; TF_CHECK_OK(NodeBuilder("A", "NoOp").Finalize(&graph, &node)); setenv("TF_DUMP_GRAPH_PREFIX", testing::TmpDir().c_str(), 1); UseMlirForGraphDump(MlirDumpConfig()); string ret = DumpGraphToFile("tir", graph);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util.cc
// dependency edge to the _SOURCE node. for (int64_t i = 0, end = args.size(); i < end; ++i) { Node* node; string arg_name = absl::StrCat("_arg", i); Status status = NodeBuilder(arg_name, FunctionLibraryDefinition::kArgOp) .ControlInput(graph->source_node()) .Attr("T", args[i].kind == XlaArgument::kResource ? DT_RESOURCE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.h
const ApiDef& api_def, const std::unordered_map<string, string>& inferred_input_attrs); struct OpInfo { // graph_op_def: The OpDef used by the runtime, has the names that // must be used when calling NodeBuilder. // interface_op_def: The OpDef used in the interface in the generated // code, with possibly overridden names and defaults. OpInfo(const OpDef& graph_op_def, const ApiDef& api_def,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 4.6K bytes - Viewed (0)