- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for XlaSpmdShardToFullShape (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); let results = (outs TF_Tensor:$output ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; } def TF_XlaSpmdShardToFullShapeOp : TF_Op<"XlaSpmdShardToFullShape", [Pure]> { let summary = [{ An op used by XLA SPMD partitioner to switch from manual partitioning to }]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)