Searched refs:GetGeneratorComputation (Results 1 – 2 of 2) sorted by relevance
56 StatusOr<HloComputation*> RngBitGeneratorExpander::GetGeneratorComputation( in GetGeneratorComputation() function in xla::RngBitGeneratorExpander114 GetGeneratorComputation(data_shape, state_shape, algorithm, module)); in ExpandInstruction()
61 StatusOr<HloComputation*> GetGeneratorComputation(const Shape& data_shape,