| /external/llvm-project/mlir/lib/IR/ |
| D | AffineExpr.cpp | 302 AffineBinaryOpExpr binaryExpr = expr.cast<AffineBinaryOpExpr>(); in isDivisibleBySymbol() local
303 return isDivisibleBySymbol(binaryExpr.getLHS(), symbolPos, opKind) && in isDivisibleBySymbol()
304 isDivisibleBySymbol(binaryExpr.getRHS(), symbolPos, opKind); in isDivisibleBySymbol()
312 AffineBinaryOpExpr binaryExpr = expr.cast<AffineBinaryOpExpr>(); in isDivisibleBySymbol() local
313 return isDivisibleBySymbol(binaryExpr.getLHS(), symbolPos, in isDivisibleBySymbol()
315 isDivisibleBySymbol(binaryExpr.getRHS(), symbolPos, in isDivisibleBySymbol()
320 AffineBinaryOpExpr binaryExpr = expr.cast<AffineBinaryOpExpr>(); in isDivisibleBySymbol() local
321 return isDivisibleBySymbol(binaryExpr.getLHS(), symbolPos, opKind) || in isDivisibleBySymbol()
322 isDivisibleBySymbol(binaryExpr.getRHS(), symbolPos, opKind); in isDivisibleBySymbol()
334 AffineBinaryOpExpr binaryExpr = expr.cast<AffineBinaryOpExpr>(); in isDivisibleBySymbol() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_common.h | 349 Assign(d, out, in0.binaryExpr(in1, typename Functor::func()));
386 Assign(dev, out, in0.binaryExpr(in1, func));
389 Assign(dev, out, in0.binaryExpr(rhs, func));
392 Assign(dev, out, lhs.binaryExpr(in1, func));
396 Assign(dev, out, lhs.binaryExpr(rhs, func));
410 Assign(d, out, in0.binaryExpr(in1, typename Functor::func()));
491 Assign(dev, out, lhs.binaryExpr(rhs, 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));
[all …]
|
| D | cwise_ops_gpu_common.cu.h | 55 To32Bit(in0).binaryExpr(in1, typename Functor::func());
93 To32Bit(in0).binaryExpr(To32Bit(in1).broadcast(bcast1), func);
98 To32Bit(in0).broadcast(bcast0).binaryExpr(To32Bit(in1), func);
102 To32Bit(out).device(d) = To32Bit(in0).broadcast(bcast0).binaryExpr(
|
| D | unravel_index_op.cc | 100 output = output.binaryExpr(strides, mod_op<Tidx>()) / strides_shifted; in Compute()
122 output = output.binaryExpr(strides.reshape(reshape).broadcast(bcast), in Compute()
|
| D | cwise_op_clip.cc | 116 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 ()()
|
| D | cwise_ops_gpu_gradients.cu.h | 47 To32Bit(in1).binaryExpr(in2, typename Functor::func());
|
| D | eigen_activations_test.cc | 85 result = input.binaryExpr(input.constant(0.01), scalar_clip_op<float>()); in TEST()
|
| D | sparse_dense_binary_op_shared.cc | 160 values_t->flat<T>().binaryExpr(dense_gathered_flat, in Compute()
|
| D | training_ops_gpu.cu.cc | 641 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
643 auto new_accum_power = new_accum.binaryExpr( in operator ()()
676 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
678 auto new_accum_power = new_accum.binaryExpr( in operator ()()
713 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
715 auto new_accum_power = new_accum.binaryExpr( in operator ()()
753 auto accum_power = accum.binaryExpr(lr_power_bcast, in operator ()()
755 auto new_accum_power = new_accum.binaryExpr( in operator ()()
|
| D | sparse_sparse_binary_op_shared.cc | 224 a_augmented_values_t.binaryExpr(b_augmented_values_t, in Compute()
|
| D | cwise_ops_gradients.h | 187 out.device(d) = in0.binaryExpr(in1, typename Functor::func());
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 140 return binaryExpr(other.derived(), internal::scalar_igamma_op<Scalar>()); in igamma()
147 return binaryExpr(other.derived(), internal::scalar_igammac_op<Scalar>()); in igammac()
154 return binaryExpr(other.derived(), internal::scalar_zeta_op<Scalar>()); in zeta()
161 return binaryExpr(other.derived(), internal::scalar_polygamma_op<Scalar>()); in polygamma()
329 binaryExpr(const OtherDerived& other, const CustomBinaryOp& func) const { in binaryExpr() function
337 return binaryExpr(other.derived(), internal::scalar_sum_op<Scalar>());
343 return binaryExpr(other.derived(), internal::scalar_difference_op<Scalar>());
349 return binaryExpr(other.derived(), internal::scalar_product_op<Scalar>());
355 return binaryExpr(other.derived(), internal::scalar_quotient_op<Scalar>());
361 return binaryExpr(other.derived(), internal::scalar_max_op<Scalar>()); in cwiseMax()
[all …]
|
| /external/eigen/doc/examples/ |
| D | class_CwiseBinaryOp.cpp | 16 cout << m1.binaryExpr(m2, MakeComplexOp<double>()) << endl; in main()
|
| /external/skia/src/sksl/ |
| D | SkSLInliner.cpp | 304 const BinaryExpression& binaryExpr = expression.as<BinaryExpression>(); in inlineExpression() local
306 expr(binaryExpr.left()), in inlineExpression()
307 binaryExpr.getOperator(), in inlineExpression()
308 expr(binaryExpr.right())); in inlineExpression()
936 BinaryExpression& binaryExpr = (*expr)->as<BinaryExpression>(); in visitExpression() local
937 this->visitExpression(&binaryExpr.left()); in visitExpression()
947 Operator op = binaryExpr.getOperator(); in visitExpression()
951 this->visitExpression(&binaryExpr.right()); in visitExpression()
|
| /external/eigen/Eigen/src/Core/ |
| D | Dot.h | 36 return a.template binaryExpr<conj_prod>(b).sum(); in run()
48 return a.transpose().template binaryExpr<conj_prod>(b).sum();
|
| /external/eigen/Eigen/src/plugins/ |
| D | CommonCwiseBinaryOps.h | 43 binaryExpr(const EIGEN_CURRENT_STORAGE_BASE_CLASS<OtherDerived> &other, const CustomBinaryOp& func …
|
| /external/eigen/doc/ |
| D | NewExpressionType.dox | 8 …er using custom \ref MatrixBase::unaryExpr "unary" or \ref MatrixBase::binaryExpr "binary" functor…
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 108 cs.binaryExpr(cs.constant(T(cell_clip)), Eigen::scalar_clip_op<T>()); in LSTMBlockCellFpropWithEigen()
|