Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.h618 llvm::Value* GetOutputAddressForReduction(
H A Dir_emitter_unnested.cc3720 llvm::Value* IrEmitterUnnested::GetOutputAddressForReduction( in GetOutputAddressForReduction() function in xla::gpu::IrEmitterUnnested
3878 llvm::Value* output_address = GetOutputAddressForReduction( in EmitReductionOutputForRowReduction()
4038 llvm::Value* output_address = GetOutputAddressForReduction( in EmitReductionOutputForColumnReduction()