Home
last modified time | relevance | path

Searched refs:unaryExpr (Results 1 – 25 of 28) sorted by relevance

12

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h71 unaryExpr(const CustomUnaryOp& func) const { in unaryExpr() function
79 return unaryExpr(internal::scalar_opposite_op<Scalar>());
85 return unaryExpr(internal::scalar_sqrt_op<Scalar>()); in sqrt()
91 return unaryExpr(internal::scalar_sign_op<Scalar>()); in sign()
97 return unaryExpr(internal::scalar_rsqrt_op<Scalar>()); in rsqrt()
103 return unaryExpr(internal::scalar_square_op<Scalar>()); in square()
109 return unaryExpr(internal::scalar_cube_op<Scalar>()); in cube()
115 return unaryExpr(internal::scalar_inverse_op<Scalar>()); in inverse()
121 return unaryExpr(internal::scalar_tanh_op<Scalar>()); in tanh()
127 return unaryExpr(internal::scalar_lgamma_op<Scalar>()); in lgamma()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_activations_test.cc37 result = input.unaryExpr(scalar_sigmoid_fast_derivative_op<float>()); in TEST()
61 result = input.unaryExpr(scalar_tanh_fast_derivative_op<float>()); in TEST()
Dcwise_ops_gpu_common.cu.h44 To32Bit(out).device(d) = To32Bit(in).unaryExpr(typename Functor::func());
65 To32Bit(out).device(d) = To32Bit(in).unaryExpr(Unary(scalar.data()));
75 To32Bit(out).device(d) = To32Bit(in).unaryExpr(Unary(scalar.data()));
Dcwise_ops_common.h362 Assign(d, out, in.unaryExpr(Unary(scalar.data())));
374 Assign(d, out, in.unaryExpr(Unary(scalar.data())));
423 Assign(d, out, in.unaryExpr(Unary(scalar.data())));
435 Assign(d, out, in.unaryExpr(Unary(scalar.data())));
568 Assign(d, out, in.unaryExpr(Unary(scalar.data(), error)));
580 Assign(d, out, in.unaryExpr(Unary(scalar.data(), error)));
602 Assign(d, out, in.unaryExpr(typename Functor::func()));
Dcast_op.h39 in_tensor.unaryExpr(LSBZeroSetter<IN_OUT, OUT_TYPE>()) \
57 in_tensor.unaryExpr(LSBZeroSetter<IN_OUT, OUT_TYPE>()) \
Dquantize_and_dequantize_op.h93 .unaryExpr(round_func) * in ClampScaleAndRound()
122 output.device(d) = (input * scale).unaryExpr(round_func) * inverse_scale; in ScaleAndRound()
Dcwise_op_clip.cc96 out_flat = in0_flat.unaryExpr(UnaryClipFunc<T>(in1_flat(0), in2_flat(0))); in operator ()()
Dquantize_op.cc281 .unaryExpr( in QuantizeSlice()
Dquantization_utils.h725 .unaryExpr(int64_right_shift_op<32>())) +
727 auto intermediate = fp_value.unaryExpr(int64_right_shift_op<fp_shift>());
Dunary_ops_composition.cc165 *out = in.unaryExpr(functor::func()); \
/external/eigen/Eigen/src/plugins/
DCommonCwiseUnaryOps.h114 EIGEN_DOC_UNARY_ADDONS(unaryExpr,unary function) in EIGEN_DOC_UNARY_ADDONS() argument
121 unaryExpr(const CustomUnaryOp& func = CustomUnaryOp()) const in EIGEN_DOC_UNARY_ADDONS()
/external/eigen/unsupported/Eigen/src/MatrixFunctions/
DMatrixExponential.h228 MatrixType A = arg.unaryExpr(MatrixExponentialScalingOp<float>(squarings));
256 MatrixType A = arg.unaryExpr(MatrixExponentialScalingOp<double>(squarings));
292 MatrixType A = arg.unaryExpr(MatrixExponentialScalingOp<long double>(squarings));
312 MatrixType A = arg.unaryExpr(MatrixExponentialScalingOp<long double>(squarings));
331 MatrixType A = arg.unaryExpr(MatrixExponentialScalingOp<long double>(squarings));
/external/eigen/doc/examples/
Dclass_CwiseUnaryOp_ptrfun.cpp18 cout << m1 << endl << "becomes: " << endl << m1.unaryExpr(ptr_fun(ramp)) << endl; in main()
Dclass_CwiseUnaryOp.cpp17 cout << m1 << endl << "becomes: " << endl << m1.unaryExpr(CwiseClampOp<double>(-0.5,0.5)) << endl; in main()
Dnullary_indexing.cpp60 …B = indexing(A, ArrayXi::LinSpaced(13,0,12).unaryExpr([](int x){return x%4;}), ArrayXi::LinSpaced… in main()
/external/eigen/blas/
Dlevel1_cplx_impl.h35 if(*incx==1) return make_vector(x,*n).unaryExpr<scalar_norm1_op>().sum();
36 else return make_vector(x,*n,std::abs(*incx)).unaryExpr<scalar_norm1_op>().sum();
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dgru_cell.cc69 ru = ru.unaryExpr(Eigen::internal::scalar_logistic_op<float>()); in GruCell()
/external/eigen/unsupported/test/
Dcxx11_tensor_expr.cpp240 mat2 = mat1.inverse().unaryExpr(&asinf); in test_functors()
241 mat3 = mat1.unaryExpr(&tanhf); in test_functors()
Dcxx11_tensor_of_float16_cuda.cu42 gpu_res_float.device(gpu_device) = gpu_float.unaryExpr(Eigen::internal::scalar_isnan_op<float>()); in test_cuda_numext()
43 …gpu_res_half.device(gpu_device) = gpu_float.cast<Eigen::half>().unaryExpr(Eigen::internal::scalar_… in test_cuda_numext()
/external/eigen/doc/
DQuickReference.dox475 It is also very simple to apply any user defined function \c foo using DenseBase::unaryExpr togethe…
477 mat1.unaryExpr(std::ptr_fun(foo));
478 mat1.unaryExpr(std::ref(foo));
479 mat1.unaryExpr([](double x) { return foo(x); });
DNewExpressionType.dox8 In most cases, a custom expression can be avoided by either using custom \ref MatrixBase::unaryExpr
DAsciiQuickReference.txt144 R = P.unaryExpr(ptr_fun(func)) // R = arrayfun(func, P) // with: scalar func(const scalar &x);
/external/eigen/Eigen/src/LU/
DPartialPivLU.h381 = lu.col(k).tail(rows-k).unaryExpr(Scoring()).maxCoeff(&row_of_biggest_in_col);
DFullPivLU.h522 .unaryExpr(Scoring())
/external/eigen/Eigen/src/QR/
DFullPivHouseholderQR.h489 .unaryExpr(Scoring())

12