- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for _XlaMerge (0.21 sec)
-
tensorflow/compiler/jit/ops/xla_ops.cc
Unlike the TensorFlow Merge op, which requires inputs of some types to be placed on the host, the _XlaMerge op can merge inputs of all types when placed on the device. This prevents the need for copy operations, in particular when an XLA cluster has int32 outputs. The _XlaMerge up does not have a value_index output that identifies the chosen input. )");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 09:08:06 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
REGISTER_KERNEL_BUILDER( Name("_XlaRun").Device(DEVICE_DEFAULT).HostMemory("key"), XlaRunOp); REGISTER_KERNEL_BUILDER(Name("_XlaMerge").Device(DEVICE_CPU), XlaMergeOp); REGISTER_KERNEL_BUILDER(Name("_XlaMerge").Device(DEVICE_GPU), XlaMergeOp); REGISTER_KERNEL_BUILDER(Name("_XlaMerge").Device(DEVICE_DEFAULT), XlaMergeOp);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
auto tf_call = NodeWith(Op("StatefulPartitionedCall"), CtrlDeps(NodeWith(Op("Identity"), Inputs(Out(0, predicated_compilation_key))))); auto merge = NodeWith(Op("_XlaMerge"), Inputs(Out(tf_call), Out(xla_run))); auto assign_var = NodeWith(Op("AssignVariableOp"), Inputs(_, Out(merge))); std::unique_ptr<Graph> graph; TF_ASSERT_OK(BuildXlaOps(root, fdef_lib, &graph));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
absl::StrCat("CheckNumerics failed for output ", oidx, "(", new_output.name(), ") from cluster ", cluster_name)); new_output = check_numerics_op; } ops::_XlaMerge xla_merge_op(s.WithOpName("merge_oidx_", oidx), Output(old_node, oidx), new_output); merged_output = merged_outputs[oidx] = xla_merge_op.output; } Node* dst = e->dst();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0)