Home
last modified time | relevance | path

Searched refs:EmitCallToNestedComputation (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc153 Status IrEmitter::EmitCallToNestedComputation( in EmitCallToNestedComputation() function in xla::gpu::IrEmitter
413 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in EmitAtomicOperationUsingCAS()
576 return EmitCallToNestedComputation(*call->to_apply(), operand_addresses, in HandleCall()
648 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in ComputeNestedElementFromAddrs()
H A Dir_emitter.h141 Status EmitCallToNestedComputation(const HloComputation& nested_computation,
H A Dir_emitter_unnested.cc2082 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in EmitSelectAndScatter()
2427 return EmitCallToNestedComputation(*desc.update_computation, in EmitScatter()
2714 return EmitCallToNestedComputation(*comparator, operands, output); in EmitSort()