Searched refs:EmitCallToNestedComputation (Results 1 – 3 of 3) sorted by relevance
224 Status IrEmitter::EmitCallToNestedComputation( in EmitCallToNestedComputation() function in xla::gpu::IrEmitter470 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in EmitAtomicOperationUsingCAS()594 return EmitCallToNestedComputation(*call->to_apply(), operand_addresses, in HandleCall()659 TF_RETURN_IF_ERROR(EmitCallToNestedComputation(computation, parameter_buffers, in ComputeNestedElement()
152 Status EmitCallToNestedComputation(const HloComputation& nested_computation,
2368 TF_RETURN_IF_ERROR(EmitCallToNestedComputation( in EmitSelectAndScatterFromMlir()2713 return EmitCallToNestedComputation(*desc.update_computation, in EmitScatter()3014 return EmitCallToNestedComputation(*comparator, operands, output); in EmitSortFromMlir()4352 TF_CHECK_OK(EmitCallToNestedComputation( in EmitFullWarpShuffleDownLoopForReduce()4657 TF_CHECK_OK(EmitCallToNestedComputation( in EmitTileElementForReduction()