- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for CollectiveReduceV2 (0.18 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-collective.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
%token: tensor<!tf_type.resource<tensor<f32>>> ) -> tensor<3xi64> { %0 = "tf_device.cluster"() ({ // arg4 shall have been removed. // CHECK: tf.CollectiveReduceV2 // CHECK-NOT: [[TOKEN]] // CHECK-SAME: merge_op %0 = "tf.CollectiveReduceV2"(%input, %group_size, %group_key, %instance_key, %token) { merge_op = "Add", final_op = "Id" } : (
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
replica_groups, all_reduce.getMode(), all_reduce.getInput(), merge_op, final_op, all_reduce); } }; // Converts CollectiveReduceV2, with or without a preceding // CollectiveAssignGroupV2. Not thread-safe. class ConvertCollectiveReduceV2 : public CollectiveRewritePattern<TF::CollectiveReduceV2Op> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
LogNotCompilable(node, uncompilable_reason); return false; } if (!op_filter_.allow_collective_reduce_v2 && node.type_string() == "CollectiveReduceV2") { absl::string_view uncompilable_reason = "Collective op"; MaybeMarkUncompilableNode(uncompilable_reason, *stack_trace, encapsulating_function, uncompilable_nodes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"LeakyReluGrad", "Elu", "EluGrad", "Selu", "SeluGrad", "Select", "SelectV2", "Transpose", "ConjugateTranspose", "_UnaryOpsComposition", "CollectiveReduceV2", "CollectiveAssignGroupV2", // The following 5 operations are converted to identity "PlaceholderWithDefault", "PreventGradient", "StopGradient",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
RELEASE.md
* Fixes an integer division by 0 in `tf.raw_ops.AllToAll` ([CVE-2021-41218](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2021-41218)) * Fixes a use after free and a memory leak in `CollectiveReduceV2` ([CVE-2021-41220](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2021-41220)) * Fixes an undefined behavior via `nullptr` reference binding in sparse matrix multiplication
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandSizeAttr Nordering_token = TF_DerivedOperandSizeAttr<4>; TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF_CollectiveReduceV2Op : TF_Op<"CollectiveReduceV2", [DeclareOpInterfaceMethods<TF_GetResourceInstanceInterface>, TF_CollectiveReduceOrderingEffect]> { let summary = [{ Mutually reduces multiple tensors of identical type and shape. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)