Searched refs:EmitCallToNestedComputation (Results 1 – 3 of 3) sorted by relevance
151 Status IrEmitter::EmitCallToNestedComputation( in EmitCallToNestedComputation() function in xla::gpu::IrEmitter371 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in EmitAtomicOperationUsingCAS()714 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in HandleReduce()740 return EmitCallToNestedComputation(*call->to_apply(), operand_addresses, in HandleCall()794 TF_RETURN_IF_ERROR(EmitCallToNestedComputation(computation, parameter_buffers, in ComputeNestedElement()
156 Status EmitCallToNestedComputation(const HloComputation& nested_computation,
918 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in HandleSelectAndScatter()1340 return EmitCallToNestedComputation(*sort->to_apply(), operands, in HandleSort()2771 TF_CHECK_OK(EmitCallToNestedComputation( in EmitFullWarpShuffleDownLoopForAllReduces()2838 TF_CHECK_OK(EmitCallToNestedComputation( in EmitEpilogueForReduction()2934 TF_CHECK_OK(EmitCallToNestedComputation( in EmitTileElementForReduction()