- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for XlaAllReduce (0.22 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-collective.mlir
%group_assignment = "tf.Const"() { value = dense<[[0],[1]]> : tensor<2x1xi32> } : () -> tensor<2x1xi32> // expected-error@+1 {{'tf.XlaAllReduce' op operand #0 must be tensor of bfloat16 or 16-bit float or 32-bit float or 32-bit integer or 32-bit unsigned integer values, but got 'tuple<tensor<f32>, tensor<f32>>'}}
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/tf2xla/internal/passes/xla_broadcast.cc
topology_attr.getValue(), status_or_device_coodinates.value()); return success(); } // Move a broadcast into the XLA cluster, converting it to an XlaAllReduce. This // skips if the element type is not known to be valid for XlaAllReduce. LogicalResult MoveBroadcastToCluster(OpBuilder& builder, OpBuilder& inner_builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
let summary = "Moves a broadcast from host into XLA, encoded as XlaAllReduce"; let description = [{ This pass moves brodcasts from host TF ops into XLA. This enables use of the inter-device network, which is faster than the inter-host network. Broadcasts in XLA are encoded as XlaAllReduce. An all_reduce with all 0 inputs except for one real input produces the same result as a broadcast.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
protected: int64_t& channel_id_; // A unique channel_id shared by all rewrite patterns // in this pass. Not thread-safe. }; // Converts XlaAllReduce. Not thread-safe. class ConvertXlaAllReduce : public CollectiveRewritePattern<TF::XlaAllReduceOp> { public: using CollectiveRewritePattern::CollectiveRewritePattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0)