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
« 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# ==============================================================================
17from tensorflow.python.framework import ops
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().")