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

Update comment

Browse Source
This commit is contained in:
Frederic Bastien 2019-07-12 06:20:56 -07:00
parent 7d22d959e0
commit 5fea6eecde
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/service/gpu/ir_emitter_unnested.h
View File

@ -228,8 +228,10 @@ class IrEmitterUnnested : public IrEmitter,
// `scatter_indices_gen`, updates from`updates_gen`. The output buffer is // `scatter_indices_gen`, updates from`updates_gen`. The output buffer is
// expected to have the operand values in it already. If use_atomic // expected to have the operand values in it already. If use_atomic
// is true, we will use an atomic update. Using false for use_atomic // is true, we will use an atomic update. Using false for use_atomic
// is safe only when it is guaranteed that there is no duplicate // is safe only when it is guaranteed that there are no duplicate
// indices. // indices.
// When using use_atomi=false, it is the caller responsibility to
// ensure there is overlap.
Status EmitScatter(Thunk* thunk, HloInstruction* scatter, Status EmitScatter(Thunk* thunk, HloInstruction* scatter,
const llvm_ir::ElementGenerator& scatter_indices_gen, const llvm_ir::ElementGenerator& scatter_indices_gen,
const llvm_ir::ElementGenerator& updates_gen, const llvm_ir::ElementGenerator& updates_gen,