Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/compiler/jit/ops/xla_ops_grad.py: 60%

5 statements  

« prev     ^ index     » next       coverage.py v7.4.0, created at 2024-01-03 07:57 +0000

1"""Gradients for XLA ops.""" 

2# Copyright 2018 The TensorFlow Authors. All Rights Reserved. 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); 

5# you may not use this file except in compliance with the License. 

6# You may obtain a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, 

12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

13# See the License for the specific language governing permissions and 

14# limitations under the License. 

15# ============================================================================== 

16 

17from tensorflow.python.framework import ops 

18 

19 

20@ops.RegisterGradient("XlaClusterOutput") 

21def _XlaClusterOutputGrad(_, grad): 

22 del grad # unused 

23 raise RuntimeError("Gradient computation of graph in xla.compile() is " 

24 "prohibited because it can cause performance degradation." 

25 "Please move gradient computation inside xla.compile().")