Searched refs:allow_mixed_precision (Results 1 – 3 of 3) sorted by relevance
32 ShapeVerifier(bool layout_sensitive, bool allow_mixed_precision) in ShapeVerifier() argument34 allow_mixed_precision_(allow_mixed_precision) {} in ShapeVerifier()200 bool layout_sensitive, bool allow_mixed_precision, in DefaultVerifierMetadata() argument204 allow_mixed_precision_(allow_mixed_precision) {} in DefaultVerifierMetadata()225 bool layout_sensitive, bool allow_mixed_precision,231 layout_sensitive, allow_mixed_precision, shape_size_func)), in target_metadata_() argument
99 bool allow_mixed_precision);
568 bool allow_mixed_precision) { in VerifyHloModule() argument570 /*allow_mixed_precision=*/allow_mixed_precision) in VerifyHloModule()