Home
last modified time | relevance | path

Searched refs:needs_upcast (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath.cc72 bool needs_upcast = absl::c_linear_search(upcast_types, elem_ty); in DoWithUpcastToF32() local
74 if (needs_upcast) { in DoWithUpcastToF32()
78 if (needs_upcast) { in DoWithUpcastToF32()
957 bool needs_upcast = in Igamma() local
960 if (needs_upcast) { in Igamma()
966 if (needs_upcast) { in Igamma()
1004 bool needs_upcast = in IgammaGradA() local
1007 if (needs_upcast) { in IgammaGradA()
1012 if (needs_upcast) { in IgammaGradA()
1051 bool needs_upcast = in RandomGammaGrad() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dchlo_legalize_to_hlo.cc417 bool needs_upcast = original_ty.getWidth() < min_precision_ty.getWidth(); in MaterializeWithUpcast() local
421 if (needs_upcast) { in MaterializeWithUpcast()
432 if (needs_upcast) { in MaterializeWithUpcast()