Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for XlaSpmdShardToFullShape (0.29 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir 

        // CHECK-SAME:          recv_key = "host_compute_channel_0_retvals"
    // CHECK-SAME:          send_key = "host_compute_channel_0_args"
    // CHECK:             %[[B_FULL:.+]] = "tf.XlaSpmdShardToFullShape"(%[[B]]) <{dim = -1 : i64, full_shape = #tf_type.shape<2x2>, manual_sharding = "\08\03\1A\02\02\01\22\02\00\01", unspecified_dims = []}> : (tensor<1x2xi64>) -> tensor<2x2xi64>
    // CHECK:             "tf.OpC"(%[[B_FULL]])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 129.6K bytes
    - Viewed (0)
Back to top