- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for tf_outputs (0.16 sec)
-
tensorflow/c/c_api.cc
cond_graph->parent_inputs = inputs; body_graph->parent = g; body_graph->parent_inputs = inputs; TF_Output* cond_inputs = new TF_Output[ninputs]; TF_Output cond_output = {nullptr, -1}; TF_Output* body_inputs = new TF_Output[ninputs]; TF_Output* body_outputs = new TF_Output[ninputs]; for (int i = 0; i < ninputs; ++i) body_outputs[i] = {nullptr, -1}; const char* name = nullptr;
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/jit/encapsulate_subgraphs_pass_test.cc
NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"), "_XlaRecvAtHost", opts.op_registry()); node_builder.Input(std::move(key_input)); return opts.WithAttr("Toutputs", dtypes) .WithAttr("key", key) .WithAttr("device_ordinal", 0) .WithAttr("_encapsulate", cluster) .WithAttr("_outside", oc_cluster) .FinalizeBuilder(&node_builder); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
NodeDefBuilder recv_at_host_builder( absl::StrCat("outside_compilation_", oc_cluster_name, "_recv"), "_XlaRecvAtHost"); NodeDef recv_at_host_def; recv_at_host_builder.Attr("Toutputs", recv_at_host_dtypes); // The correct device_ordinal will be inserted during replication in a // subsequent rewrite. AttrValue device_ordinal_value; device_ordinal_value.set_placeholder("_device_ordinal");
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/tests/canonicalize.mlir
"tf._SomeOp"(%0#1, %0#0) : (tensor<!tf_type.resource>, tensor<f32>) -> () // CHECK: "tf._SomeOp"(%[[ARG2]], %[[IF_OUTPUT]]) : (tensor<!tf_type.resource>, tensor<f32>) -> () "tf._SomeOp"(%0#3, %0#2) : (tensor<!tf_type.resource>, tensor<f32>) -> () // CHECK: return %[[IF_OUTPUT]] : tensor<f32> func.return %0#2 : tensor<f32> } // CHECK-LABEL: func @eliminatePassThroughCaseRegion(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
// CHECK: "tf._XlaSendFromHostV2"(%[[E_OUTPUT]], %[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]]) // CHECK: %[[F_OUTPUT:[0-9]*]] = "tf.F"(%[[C_OUTPUT]]#1) // CHECK: "tf._XlaSendFromHostV2"(%[[F_OUTPUT]], %[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]]) // CHECK: %[[RECV_OUTPUT2:[0-9]*]]:2 = "tf._XlaRecvAtHostV2"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 129.6K bytes - Viewed (0)