Searched refs:GetMlirEmitterInput (Results 1 – 2 of 2) sorted by relevance
630 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(hlo)); in DefaultAction()737 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(constant)); in HandleConstant()1063 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(custom_call)); in HandleCustomCall()1565 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(fft)); in HandleFft()1594 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(hlo)); in HandleTriangularSolve()1742 StatusOr<MlirEmitterInput> IrEmitterUnnested::GetMlirEmitterInput( in GetMlirEmitterInput() function in xla::gpu::IrEmitterUnnested1874 TF_ASSIGN_OR_RETURN(auto mlir_input, GetMlirEmitterInput(fusion)); in HandleFusion()2083 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(copy)); in HandleCopy()2138 TF_ASSIGN_OR_RETURN(auto mlir_input, GetMlirEmitterInput(reduce)); in HandleReduce()2195 TF_ASSIGN_OR_RETURN(auto input, GetMlirEmitterInput(select_and_scatter)); in HandleSelectAndScatter()[all …]
731 StatusOr<MlirEmitterInput> GetMlirEmitterInput(HloInstruction* hlo);