experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Comment that output_shape_index can be set to negative for non-updated variables.

Browse Source 
PiperOrigin-RevId: 269695929
This commit is contained in:
Yuanzhong Xu 2019-09-17 17:58:16 -07:00 committed by TensorFlower Gardener
parent 143101a4fc
commit d3b65857f2
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/xla/service/hlo_module_config.h
View File

@ -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).