Home
last modified time | relevance | path

Searched refs:binaryExpr (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_common.h364 Assign(d, out, in0.binaryExpr(in1, typename Functor::func()));
401 Assign(dev, out, in0.binaryExpr(in1, func));
404 Assign(dev, out, in0.binaryExpr(rhs, func));
407 Assign(dev, out, lhs.binaryExpr(in1, func));
411 Assign(dev, out, lhs.binaryExpr(rhs, func));
425 Assign(d, out, in0.binaryExpr(in1, typename Functor::func()));
497 Assign(dev, out, lhs.binaryExpr(rhs, func));
503 Assign(dev, out, lhs.binaryExpr(rhs, func));
509 Assign(dev, out, lhs.binaryExpr(rhs, func));
515 Assign(dev, out, lhs.binaryExpr(rhs, func));
[all …]
Dcwise_ops_gpu_common.cu.h60 out32.device(d) = in0_32.binaryExpr(in1_32, typename Functor::func());
110 in0_32.binaryExpr(in1_32.broadcast(bcast1), func);
119 in0_32.broadcast(bcast0).binaryExpr(in1_32, func);
127 out32.device(d) = in0_32.broadcast(bcast0).binaryExpr(
Dunravel_index_op.cc132 output = output.binaryExpr(strides, mod_op<Tidx>()) / strides_shifted; in Compute()
154 output = output.binaryExpr(strides.reshape(reshape).broadcast(bcast), in Compute()
Dcwise_op_clip.cc116 in0_flat.binaryExpr(in2_flat, BinaryRightClipFunc<T>(in1_flat(0))); in operator ()()
136 in0_flat.binaryExpr(in1_flat, BinaryLeftClipFunc<T>(in2_flat(0))); in operator ()()
Dcwise_ops_gpu_gradients.cu.h48 out32.device(d) = in1_32.binaryExpr(in2, typename Functor::func());
Deigen_activations_test.cc85 result = input.binaryExpr(input.constant(0.01), scalar_clip_op<float>()); in TEST()
Dtraining_ops_gpu.cu.cc568 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
570 auto new_accum_power = new_accum.binaryExpr( in operator ()()
603 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
605 auto new_accum_power = new_accum.binaryExpr( in operator ()()
640 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
642 auto new_accum_power = new_accum.binaryExpr( in operator ()()
680 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
682 auto new_accum_power = new_accum.binaryExpr( in operator ()()
Dsparse_dense_binary_op_shared.cc189 values_t->flat<T>().binaryExpr(dense_gathered_flat, in Compute()
Dsparse_sparse_binary_op_shared.cc232 a_augmented_values_t.binaryExpr(b_augmented_values_t, in Compute()
Dcwise_ops_gradients.h182 out.device(d) = in0.binaryExpr(in1, typename Functor::func());
/external/skia/src/sksl/ir/
DSkSLFunctionDefinition.cpp168 BinaryExpression& binaryExpr = nextExpr->as<BinaryExpression>(); in Convert() local
169 if (binaryExpr.getOperator().kind() != OperatorKind::EQ) { in Convert()
173 Expression& leftExpr = *binaryExpr.left(); in Convert()
184 if (Analysis::ContainsVariable(*binaryExpr.right(), *varRef.variable())) { in Convert()
189 vardecl->value() = std::move(binaryExpr.right()); in Convert()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h214 return binaryExpr(other.derived(), internal::scalar_igamma_op<Scalar>()); in igamma()
221 return binaryExpr(other.derived(), internal::scalar_igamma_der_a_op<Scalar>()); in igamma_der_a()
228 return binaryExpr(other.derived(), internal::scalar_gamma_sample_der_alpha_op<Scalar>()); in gamma_sample_der_alpha()
235 return binaryExpr(other.derived(), internal::scalar_igammac_op<Scalar>()); in igammac()
242 return binaryExpr(other.derived(), internal::scalar_zeta_op<Scalar>()); in zeta()
249 return binaryExpr(other.derived(), internal::scalar_polygamma_op<Scalar>()); in polygamma()
454 binaryExpr(const OtherDerived& other, const CustomBinaryOp& func) const { in binaryExpr() function
462 return binaryExpr(other.derived(), internal::scalar_sum_op<Scalar>());
468 return binaryExpr(other.derived(), internal::scalar_difference_op<Scalar>());
474 return binaryExpr(other.derived(), internal::scalar_product_op<Scalar>());
[all …]
/external/eigen/doc/examples/
Dclass_CwiseBinaryOp.cpp16 cout << m1.binaryExpr(m2, MakeComplexOp<double>()) << endl; in main()
/external/skia/src/sksl/
DSkSLInliner.cpp210 const BinaryExpression& binaryExpr = expression.as<BinaryExpression>(); in inlineExpression() local
213 expr(binaryExpr.left()), in inlineExpression()
214 binaryExpr.getOperator(), in inlineExpression()
215 expr(binaryExpr.right())); in inlineExpression()
889 BinaryExpression& binaryExpr = (*expr)->as<BinaryExpression>(); in visitExpression() local
890 this->visitExpression(&binaryExpr.left()); in visitExpression()
900 Operator op = binaryExpr.getOperator(); in visitExpression()
904 this->visitExpression(&binaryExpr.right()); in visitExpression()
/external/eigen/Eigen/src/Core/
DDot.h37 return a.template binaryExpr<conj_prod>(b).sum(); in run()
50 return a.transpose().template binaryExpr<conj_prod>(b).sum();
/external/eigen/Eigen/src/plugins/
DCommonCwiseBinaryOps.h43 binaryExpr(const EIGEN_CURRENT_STORAGE_BASE_CLASS<OtherDerived> &other, const CustomBinaryOp& func …
/external/eigen/doc/
DNewExpressionType.dox8 …er using custom \ref MatrixBase::unaryExpr "unary" or \ref MatrixBase::binaryExpr "binary" functor…
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.cc108 cs.binaryExpr(cs.constant(T(cell_clip)), Eigen::scalar_clip_op<T>()); in LSTMBlockCellFpropWithEigen()