From d3b65857f25266a717384f979dc6b4e98d28d167 Mon Sep 17 00:00:00 2001 From: Yuanzhong Xu Date: Tue, 17 Sep 2019 17:58:16 -0700 Subject: [PATCH] Comment that output_shape_index can be set to negative for non-updated variables. PiperOrigin-RevId: 269695929 --- tensorflow/compiler/xla/service/hlo_module_config.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/compiler/xla/service/hlo_module_config.h b/tensorflow/compiler/xla/service/hlo_module_config.h index dddaa3fd259..7dd143b95b0 100644 --- a/tensorflow/compiler/xla/service/hlo_module_config.h +++ b/tensorflow/compiler/xla/service/hlo_module_config.h @@ -41,6 +41,9 @@ class HloModuleConfig { // programs, and the caller is responsible to call the XLA-generated // sharding/unsharding programs before and after the sharded main program. // + // If the variable is not updated and there is not a corresponding output, use + // {-1} as the output_shape_index. + // // The sharding/unsharding programs will include all the input/output pairs in // shardable_value_update_pairs() as a flat tuple in their inputs/outputs, // sorted by (input_parameter_number, parameter_shape_index).