diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-23 12:48:00 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-23 12:51:41 -0800 |
commit | 0eba6b6b6a0a0f0d56afd3e2c6945075706f9abb (patch) | |
tree | 81338b45a592dba21f108fa4f2494309a05af623 /tensorflow/compiler/xla/service/hlo_matchers.h | |
parent | 9f389fc4d36f0ffb79681817c25b8d7a329fc649 (diff) |
[XLA:GPU] Support BF16 data type.
Add an HLO pass to the GPU backend to implement BF16 operations with F32
operations.
Define macro XLA_BACKEND_SUPPORTS_BFLOAT16=1 when building tests for the GPU
backend to enable BF16 tests for GPU. Enable bfloat16_test and other BF16 tests
for GPU.
Add hlo_element_type_converter_test.
Add convolution tests and matrix multiplication tests for BF16.
PiperOrigin-RevId: 182977358
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_matchers.h')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_matchers.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_matchers.h b/tensorflow/compiler/xla/service/hlo_matchers.h index 992f55788b..9206cdac05 100644 --- a/tensorflow/compiler/xla/service/hlo_matchers.h +++ b/tensorflow/compiler/xla/service/hlo_matchers.h @@ -83,6 +83,7 @@ HLO_MATCHER(Abs); HLO_MATCHER(Add); HLO_MATCHER(Bitcast); HLO_MATCHER(Broadcast); +HLO_MATCHER(BatchNormGrad); HLO_MATCHER(Call); HLO_MATCHER(Ceil); HLO_MATCHER(Clamp); |