- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for host_mlir_module (0.2 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
loc, shard_output_types, manual_inputs, /*send_key=*/builder.getStringAttr(args_communication_key), /*recv_key=*/builder.getStringAttr(retvals_communication_key), /*host_mlir_module=*/builder.getStringAttr(serialized_func_module), /*manual_sharding=*/builder.getBoolAttr(true)); host_compute_out_ops.push_back(host_compute); // Convert MANUAL sharded outputs to split sharded outputs.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0)