xla

c3421393 - Update isnan tensor method to cast to xla::PrimitiveType::PRED (#3242)

Update isnan tensor method to cast to xla::PrimitiveType::PRED (#3242) * Update isnan tensor method to cast to xla::PrimitiveType::PRED * Rebase with master * Fix build issue