Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h34 allow_mixed_precision_(allow_mixed_precision) {} in ShapeVerifier()
138 return allow_mixed_precision_ in SameElementType()
168 bool allow_mixed_precision_; variable
204 allow_mixed_precision_(allow_mixed_precision) {} in DefaultVerifierMetadata()
212 allow_mixed_precision_); in GetVerifier()
217 bool allow_mixed_precision_; variable
Dhlo_verifier.cc277 (allow_mixed_precision_ && in HasCompatibleElementTypes()
457 return allow_mixed_precision_ in HandleReduce()
616 return allow_mixed_precision_ in HandleMap()
630 return allow_mixed_precision_ in HandleReduceWindow()
851 if (!allow_mixed_precision_) { in CheckShape()
887 if (allow_mixed_precision_) { in CheckShape()