- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for output_sharding_config (0.5 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
location, cluster_func_output_index, output_sharding_config, cluster_to_core_index, cluster_idx, new_parallel_execute, &outputs_to_merge); if (failed(status)) return mlir::failure(); // Creates a tree of Concat ops that merges outputs from multiple logical // devices to a single replica output. const xla::OpSharding& sharding = output_sharding_config[cluster_func_output_index];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
// different outputs depending on the output sharding configuration. mlir::LogicalResult GetOutputTypesForLogicalDeviceComputation( int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, mlir::tf_device::ClusterFuncOp cluster_func, llvm::SmallVectorImpl<mlir::Type>* output_types, llvm::SmallVectorImpl<int>* cluster_to_core_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
} // Creates a `tf.TPUExecute` op that executes TPU program. LogicalResult BuildExecuteOp( const int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, llvm::ArrayRef<Value> inputs, tf_device::ClusterFuncOp cluster_func, OpBuilder* builder, TF::TPUExecuteOp* execute_op) { // TODO(b/139377366): Need to snapshot all resource variable inputs in
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)