/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...] |