| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | gpu_op_digamma.cc | 21 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Digamma, DT_HALF);
22 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Digamma, DT_FLOAT);
23 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Digamma, DT_DOUBLE);
|
| D | gpu_unary_ops_test.cc | 241 Digamma, DT_FLOAT, DT_DOUBLE, DT_FLOAT, DT_DOUBLE,
246 Digamma, DT_DOUBLE, DT_DOUBLE, test::InputAsVector<double>(kDigammaValues),
250 Digamma, DT_HALF, DT_DOUBLE, DT_HALF, DT_DOUBLE,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Digamma.pbtxt | 2 name: "Digamma"
24 name: "Digamma"
47 name: "Digamma"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Digamma.pbtxt | 2 name: "Digamma"
24 name: "Digamma"
47 name: "Digamma"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_Digamma.pbtxt | 2 graph_op_name: "Digamma"
4 name: "math.Digamma"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_Digamma.pbtxt | 2 graph_op_name: "Digamma"
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/op_definitions/ |
| D | digamma.mlir.tmpl | 3 %0 = "tf.Digamma"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Digamma.pbtxt | 2 graph_op_name: "Digamma"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unary_ops.cc | 132 XLAJIT_MAKE_UNARY(Digamma, xla::Digamma(x));
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math_test.cc | 213 {Digamma, "digamma"}, in XLA_TEST_F()
393 XLA_TEST_F(MathTest, Digamma) { in XLA_TEST_F() argument
397 Digamma(x); in XLA_TEST_F()
567 Digamma(x); in XLA_TEST_F()
|
| D | math.h | 59 XlaOp Digamma(XlaOp input);
|
| D | math.cc | 646 XlaOp Digamma(XlaOp input) { in Digamma() function
779 XlaOp dlogax_da = Log(x) - Digamma(a + ScalarLike(a, 1)); in IgammaSeries()
912 XlaOp dlogax_da = Log(x) - Digamma(a); in IgammacContinuedFraction()
1866 XlaOp output = Select(Eq(n, ScalarLike(n, 0.)), Digamma(x), in Polygamma()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | exhaustive_unary_test_f32_or_smaller.cc | 588 UNARY_TEST_FLOAT_32_BITS_OR_LESS(Digamma, {
612 Run(Digamma, host_digamma_with_gpu_ftz_errors, error_spec_gen);
614 Run(Digamma, HostDigamma, error_spec_gen);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | chlo_ops.td | 578 let summary = "Digamma function";
581 Returns `Digamma(operand)` element-wise.
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 365 UNARY_OP(Digamma); in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 809 auto dx = Mul(grad_scope, grad, Digamma(grad_scope, x)); in LgammaGrad()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 10398 name: "Digamma"
10419 name: "Digamma"
10441 name: "Digamma"
|
| D | ops_history.v2.pbtxt | 20750 name: "Digamma"
20772 name: "Digamma"
20795 name: "Digamma"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1232 name: "Digamma"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1232 name: "Digamma"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 21873 func Digamma(scope *Scope, x tf.Output) (y tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13091 name: "Digamma"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3405 def TF_DigammaOp : TF_Op<"Digamma", [NoSideEffect, TF_SameOperandsAndResultTypeResolveRef]> {
|