| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_softmax.h | 38 SoftmaxOp(const float beta) : beta_(beta) {}
72 beta_)
80 const float beta_; member in struct:Eigen::__anon45378::SoftmaxOp
|
| lrn_op.cc | 47 // When the depth is large and beta_ is 0.5 or 1.0, Single-threaded
78 : depth_radius_(depth_radius), bias_(bias), alpha_(alpha), beta_(beta) {}
92 (beta_ == T(0.5) || beta_ == T(1))) {
107 if (beta_ == T(1)) {
110 } else if (beta_ == T(0.5)) {
114 in_shaped * (tmp.log() * -beta_).exp();
151 if (beta_ == T(1)) {
153 } else if (beta_ == T(0.5)) {
157 data_in.array() * (data_out.array().log() * -beta_).exp()
164 T beta_; member in struct:tensorflow::LaunchLRN
225 T beta_; member in struct:tensorflow::LaunchLRN
283 T beta_; member in class:tensorflow::LRNOp
381 T beta_; member in struct:tensorflow::LaunchLRNGrad
446 T beta_; member in struct:tensorflow::LaunchLRNGrad
506 T beta_; member in class:tensorflow::LRNGradOp
[all...] |
| mkl_lrn_op.cc | 83 OP_REQUIRES_OK(context, context->GetAttr("beta", &beta_));
145 kernel_size, new_alpha, beta_, bias_);
256 if (beta_ == T(1)) {
259 } else if (beta_ == T(0.5)) {
263 in_shaped * (tmp.log() * -beta_).exp();
315 float beta_; member in class:tensorflow::MklLRNOp
332 OP_REQUIRES_OK(context, context->GetAttr("beta", &beta_));
403 new_alpha, beta_, bias_);
408 kernel_size, new_alpha, beta_, bias_);
590 T dyi = T(-2) * alpha_ * beta_ * in_shaped(i, k)
671 float beta_; member in class:tensorflow::MklLRNGradOp
[all...] |
| mkl_relu_op.cc | 453 : OpKernel(context), alpha_(alpha), beta_(beta) {}
485 beta_);
548 float beta_; member in class:tensorflow::MklReluOpBase
558 : OpKernel(context), alpha_(alpha), beta_(beta) {}
642 beta_);
716 float beta_; member in class:tensorflow::MklReluGradOpBase
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gemm_thunk.h | 74 const double beta_; member in class:xla::gpu::GemmThunk
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| softmax_test.cc | 35 : batches_(batches), input_size_(size), beta_(beta) {
39 CreateSoftmaxOptions(builder_, beta_).Union());
59 float beta_; member in class:tflite::__anon46096::SoftmaxOpModel
|
| /external/webrtc/webrtc/modules/remote_bitrate_estimator/ |
| aimd_rate_control.h | 80 float beta_; member in class:webrtc::AimdRateControl
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| lrn_ops.cc | 34 OP_REQUIRES_OK(ctx, ctx->GetAttr("beta", &beta_));
64 xla::ConstantR0<float>(builder, -beta_));
73 float beta_; member in class:tensorflow::__anon44146::LRNOp
86 OP_REQUIRES_OK(ctx, ctx->GetAttr("beta", &beta_));
153 xla::Mul(xla::ConstantR0<float>(builder, -2.0f * alpha_ * beta_),
168 xla::Pow(norm, xla::ConstantR0<float>(builder, -beta_))));
177 float beta_; member in class:tensorflow::__anon44146::LRNGradOp
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | 627 : batches_(batches), input_size_(size), beta_(beta) {
631 CreateSoftmaxOptions(builder_, beta_).Union());
651 float beta_; member in class:tflite::__anon45929::SoftmaxOpModel
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 869 beta_ = beta;
891 float beta() const { return beta_; }
899 float beta_; member in class:stream_executor::dnn::NormalizeDescriptor
[all...] |
| /external/webp/src/enc/ |
| vp8i_enc.h | 193 int beta_; // filter-susceptibility, range [0,255]. member in struct:__anon49465
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.cc | 905 double beta_; member in class:stream_executor::gpu::ScopedActivationDescriptor
[all...] |
