Searched refs:igamma_grad_a (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_igammas.cc | 21 REGISTER2(BinaryOp, CPU, "IgammaGradA", functor::igamma_grad_a, float, double);
25 REGISTER2(BinaryOp, GPU, "IgammaGradA", functor::igamma_grad_a, float, double);
|
| D | cwise_op_gpu_igammas_float.cu.cc | 24 DEFINE_BINARY1(igamma_grad_a, float);
|
| D | cwise_op_gpu_igammas_double.cu.cc | 24 DEFINE_BINARY1(igamma_grad_a, double);
|
| D | cwise_ops_gradients.h | 210 struct igamma_grad_a : base<T, Eigen::internal::scalar_igamma_der_a_op<T>> {};
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 206 igamma_grad_a = _broadcasting_binary_op(gen_math_ops.igamma_grad_a) variable
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 1077 partial_a = gen_math_ops.igamma_grad_a(a, x)
1089 igamma_grad_a, igamma_grad_x = _IgammaGrad(op, grad)
1090 return (-igamma_grad_a, -igamma_grad_x)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 172 ("IgammaGradA", math_ops.igamma_grad_a),
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | math_test.py | 191 math_ops.igamma_grad_a,
|
| D | pfor.py | 2838 @RegisterPForWithArgs("IgammaGradA", math_ops.igamma_grad_a)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | special_math_test.py | 71 return -gen_math_ops.igamma_grad_a(a, x) / prob
|