Searched refs:mul_no_nan (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_mul_1.cc | 22 REGISTER6(BinaryOp, CPU, "MulNoNan", functor::mul_no_nan, Eigen::half, float,
53 REGISTER5(BinaryOp, GPU, "MulNoNan", functor::mul_no_nan, Eigen::half, float,
|
| D | cwise_op_gpu_mul.cu.cc | 29 DEFINE_BINARY5(mul_no_nan, Eigen::half, float, double, complex64, complex128);
|
| D | cwise_ops.h | 1019 struct mul_no_nan : base<T, Eigen::internal::mul_no_nan_op<T>> {};
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad_test.py | 461 outputs = math_ops.mul_no_nan(inputs, 1 + math_ops.abs(inputs))
474 outputs = math_ops.mul_no_nan(x, y)
|
| D | math_grad.py | 1395 return gen_math_ops.mul_no_nan(grad, y), gen_math_ops.mul_no_nan(x, grad)
1401 math_ops.reduce_sum(gen_math_ops.mul_no_nan(grad, y), rx), sx),
1403 math_ops.reduce_sum(gen_math_ops.mul_no_nan(x, grad), ry), sy))
|
| D | math_ops.py | 1451 return gen_math_ops.mul_no_nan(x, y, name=name)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | lower_tf.mlir | 76 // CHECK-LABEL: func @mul_no_nan
78 func @mul_no_nan(%arg0: tensor<2x3xf32>, %arg1: tensor<3xf32>) -> tensor<2x3xf32> {
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 114 Output mul_no_nan = ops::MulNoNan(s.WithOpName("mul_no_nan"), d, d); in TEST_F() local
116 Output id2 = ops::Identity(s.WithOpName("id2"), mul_no_nan); in TEST_F()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2860 @RegisterPForWithArgs("MulNoNan", math_ops.mul_no_nan)
|