- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _Arg (0.03 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} // Builds XlaRecvAtHost node, and replaces all _Arg nodes with it. absl::StatusOr<Node*> ReplaceArgNodesWithRecvAtHostNode( Graph* g, const string& oc_cluster_name, std::vector<DataType>* recv_at_host_dtypes, Node* key_placeholder) { // TODO(b/77601805): use out nodes for source node, instead of traversing all // nodes. std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
return errors::InvalidArgument( "Invalid \"_handle_dtypes\" attribute value for _Arg node: ", shape_attr->DebugString()); } if (shape_attr->list().shape().empty()) { return errors::InvalidArgument( "Invalid \"_handle_shapes\" attribute value for _Arg node: ", shape_attr->DebugString()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
node_builder.Input(inputs); return opts.FinalizeBuilder(&node_builder); } Node* ArgOp(int index, DataType type, const GraphDefBuilder::Options& opts) { return ops::SourceOp("_Arg", opts.WithAttr("T", type).WithAttr("index", index)); } Node* RetOp(int index, ops::NodeOut a, const GraphDefBuilder::Options& opts) { if (opts.HaveError()) return nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)