Home
last modified time | relevance | path

Searched refs:EmitScatter (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h468 Status EmitScatter(Thunk* thunk, mlir::lmhlo::ScatterOp scatter,
493 Status EmitScatter(const ScatterDescriptor& desc, Thunk* thunk);
Dir_emitter_unnested.cc1979 TF_RETURN_IF_ERROR(EmitScatter(desc, thunks.back().get())); in HandleFusion()
2539 TF_RETURN_IF_ERROR(EmitScatter( in EmitScatterFromMlir()
2564 Status IrEmitterUnnested::EmitScatter( in EmitScatter() function in xla::gpu::IrEmitterUnnested
2591 return EmitScatter(desc, thunk); in EmitScatter()
2594 Status IrEmitterUnnested::EmitScatter(const ScatterDescriptor& desc, in EmitScatter() function in xla::gpu::IrEmitterUnnested