HomeSort by relevance Sort by last modified time
    Searched refs:SupportsMixedPrecisions (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
bfloat16_support.h 40 virtual bool SupportsMixedPrecisions(const HloInstruction& hlo) const;
bfloat16_normalization.h 84 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override {
bfloat16_conversion_folding.cc 133 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo)) {
170 !bfloat16_support_->SupportsMixedPrecisions(*hlo)) {
188 if (!bfloat16_support_->SupportsMixedPrecisions(*crs)) {
bfloat16_support.cc 61 bool BFloat16Support::SupportsMixedPrecisions(const HloInstruction& hlo) const {
bfloat16_normalization.cc 192 if (bfloat16_support_->SupportsMixedPrecisions(*hlo)) {
208 (bfloat16_support_->SupportsMixedPrecisions(*hlo) || f32_count == 0 ||
331 if (bfloat16_support_->SupportsMixedPrecisions(*hlo) || bf16_count == 0 ||
bfloat16_conversion_folding_test.cc 58 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override {
bfloat16_propagation.cc 40 if (!bfloat16_support_->SupportsMixedPrecisions(*fusion)) {
390 if (!bfloat16_support_->SupportsMixedPrecisions(*hlo) &&
    [all...]
bfloat16_normalization_test.cc 63 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override {
bfloat16_propagation_test.cc 48 bool SupportsMixedPrecisions(const HloInstruction& hlo) const override {
    [all...]

Completed in 229 milliseconds