- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 51 for _Arg (0.07 sec)
-
tensorflow/compiler/jit/pjrt_compile_util_test.cc
namespace { StatusOr<std::unique_ptr<Graph>> SampleGraphAddXY() { std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); Scope scope = Scope::NewRootScope().ExitOnError(); auto a = ops::_Arg(scope.WithOpName("A"), DT_INT32, 0); auto b = ops::_Arg(scope.WithOpName("B"), DT_INT32, 1); auto c = ops::Add(scope.WithOpName("C"), a, b); auto d = ops::_Retval(scope.WithOpName("D"), c, 0); TF_RETURN_IF_ERROR(scope.ToGraph(graph.get()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 21 23:21:57 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/graph.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Oct 16 02:25:37 UTC 2021 - 963 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/unique_output_name.mlir
} func.return %0#0, %0#1 : tensor<2x2x1x2xi32>, tensor<2x2x1x2xf32> } } // CHECK: name: "serving_default_x" // CHECK-NEXT: op: "_Arg" // CHECK: name: "tf.MaxPoolWithArgmax" // CHECK-NEXT: op: "MaxPoolWithArgmax" // CHECK-NEXT: input: "serving_default_x" // CHECK: name: "PartitionedCall" // CHECK-NEXT: op: "_Retval"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
// "add" = "arg0" + "arg1" // "ret0" = "add" // "ret1" = "arg1" tensorflow::Scope s = tensorflow::Scope::NewRootScope(); Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_INT32, 0); Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_FLOAT, 1); Output arg2 = ops::_Arg(s.WithOpName("arg2"), DT_INT32, 2); Output add = ops::Add(s.WithOpName("add"), arg0, arg0); auto ret0 = ops::_Retval(s.WithOpName("ret0"), add, 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/end2end/disallow_stateful_partitioned_call.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 24 20:05:09 UTC 2023 - 2.6K bytes - Viewed (0) -
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/jit/encapsulate_util.h
// lifted out of If/While/function node. Attribute value will always be boolean // value "true". extern const char kXlaIsLiftedArgAttrName[]; // Attribute indicating that this node is a Placeholder node for an _Arg node // lifted out of If/While/function node. Attribute value will be a string, which // is the outside compilation cluster name sending the lifted arg node to host. extern const char kXlaLiftedArgOutsideCompilationAttrName[];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
int GetResultIndexForEdge(const Edge* edge) const; // Creates an _Arg node for the src node of edge, and add its index to // args_by_src_, if none exists yet. Also adds its index to args_by_dst_, // and adds the edge within the subgraph from the _Arg node to the image of // the dst node. Status RecordArg(const Edge* edge,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.h
// TODO(phawkins): this code does not infer accurate shapes for cyclic graphs. // `arg_shapes`: user given map from the `index` to shapes of this // node, where `index` is the `index` attribute of `_Arg` op or `_index` // attribute of `Placeholder` op. Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, const tensorflow::FunctionLibraryDefinition* fnlib_def,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
graph.SetConstructionContext(ConstructionContext::kEagerRuntime); tensorflow::set_tf2_execution(true); ConfigProto config = ConfigProto(); Scope root = Scope::NewRootScope().ExitOnError(); Output a = ops::_Arg(root.WithOpName("A"), DT_RESOURCE, 0); std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); Node* call; NameAttrList f_name_attr; f_name_attr.set_name(fd.signature().name()); TF_ASSERT_OK(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0)