Home
last modified time | relevance | path

Searched refs:bias_ (Results 1 – 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc40 void FullyConnectedOpBuilder::SetBias(TfLiteTensor* bias) { bias_ = bias; } in SetBias()
73 if (bias_ != nullptr) { in FillCoreMLBias()
75 if (bias_->type == kTfLiteFloat32) { in FillCoreMLBias()
76 const float* bias_data = GetTensorData<float>(bias_); in FillCoreMLBias()
77 std::copy(bias_data, bias_data + NumElements(bias_), in FillCoreMLBias()
81 } else if (bias_->type == kTfLiteFloat16) { in FillCoreMLBias()
86 ->assign(bias_->data.raw, bias_->bytes); in FillCoreMLBias()
Dconvolution_op_builder.cc41 void ConvolutionOpBuilder::SetBias(TfLiteTensor* bias) { bias_ = bias; } in SetBias()
194 if (bias_ != nullptr) { in FillCoreMLBias()
196 if (bias_->type == kTfLiteFloat32) { in FillCoreMLBias()
197 std::copy(bias_->data.f, bias_->data.f + NumElements(bias_->dims), in FillCoreMLBias()
201 } else if (bias_->type == kTfLiteFloat16) { in FillCoreMLBias()
206 ->assign(bias_->data.raw, bias_->bytes); in FillCoreMLBias()
/external/tensorflow/tensorflow/core/kernels/
Dlrn_op.cc86 : depth_radius_(depth_radius), bias_(bias), alpha_(alpha), beta_(beta) {} in LaunchLRN()
114 auto tmp = in_shaped.square().contract(multiplier, dims) * alpha_ + bias_; in launch()
154 data_out(i, r) = bias_ + accumulated_scale; in SingleThreadedLRN()
170 T bias_; member
180 : depth_radius_(depth_radius), bias_(bias), alpha_(alpha), beta_(beta) {} in LaunchLRN()
194 context, bias_ >= 1e-5, in launch()
195 errors::InvalidArgument("cuDNN requires bias >= 1e-5, got: ", bias_)); in launch()
211 normalize_desc.set_bias(bias_) in launch()
264 normalize_desc.set_bias(bias_) in launch()
297 T bias_; member
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlrn_ops.cc32 OP_REQUIRES_OK(ctx, ctx->GetAttr("bias", &bias_)); in LRNOp()
62 xla::Add(xla::ConstantR0<float>(builder, bias_), in Compile()
71 float bias_; member in tensorflow::__anonee6c1e180111::LRNOp
84 OP_REQUIRES_OK(ctx, ctx->GetAttr("bias", &bias_)); in LRNGradOp()
149 xla::Add(xla::ConstantR0<float>(builder, bias_), in Compile()
175 float bias_; member in tensorflow::__anonee6c1e180111::LRNGradOp
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc65 bias_ = AddInput(bias); in QuantizedTransposeConvOpModel()
70 bias_ = AddInput(bias); in QuantizedTransposeConvOpModel()
90 QuantizeAndPopulate<int32_t>(bias_, bias); in SetBias()
92 PerChannelQuantizeBias(bias_, bias); in SetBias()
95 auto* bias_tensor = interpreter_->tensor(bias_); in SetBias()
110 int bias_; member in tflite::QuantizedTransposeConvOpModel
Dmatmul_test.cc41 bias_ = AddNullInput(); in FullyConnectedOpModel()
45 bias_ = AddInput(bias); in FullyConnectedOpModel()
65 auto* bias_tensor = interpreter_->tensor(bias_); in FullyConnectedOpModel()
71 QuantizeAndPopulate<int>(bias_, data); in SetBias()
98 int bias_; member in tflite::FullyConnectedOpModel
Dconv_test.cc68 bias_ = AddInput(bias); in QuantizedConvolutionOpModel()
73 bias_ = AddInput(bias); in QuantizedConvolutionOpModel()
97 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in QuantizedConvolutionOpModel()
114 QuantizeAndPopulate<int>(bias_, data); in SetBias()
140 PerChannelQuantizeBias(bias_, data); in SetPerChannelQuantizedBias()
144 PerChannelQuantizeBias(bias_, data); in SetPerChannelQuantizedBias()
150 int bias_; member in tflite::QuantizedConvolutionOpModel
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_hybrid_test.cc64 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
91 bias_ = AddInput(bias); in BaseDepthwiseConvolutionOpModel()
96 bias_ = AddInput(bias); in BaseDepthwiseConvolutionOpModel()
117 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in BaseDepthwiseConvolutionOpModel()
123 int bias_; member in tflite::__anon4268b7900111::BaseDepthwiseConvolutionOpModel
141 PopulateTensor(bias_, data); in SetBias()
152 void SetBias(const std::vector<float>& data) { PopulateTensor(bias_, data); } in SetBias()
Dfully_connected_test.cc202 bias_ = AddNullInput();
204 bias_ = AddInput({TensorType_FLOAT32, {units_}});
219 bias_ = AddInput(bias);
223 bias_ = AddInput(bias);
240 (bias_ == kTfLiteOptionalTensor) ? std::vector<int>()
241 : GetShape(bias_)});
251 int bias_; member in tflite::__anon1e118d2b0111::BaseFullyConnectedOpModel
264 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias()
298 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
300 QuantizeAndPopulate<int64_t>(bias_, data); in SetBias()
[all …]
Dsvdf_test.cc144 bias_ = AddNullInput(); in BaseSVDFOpModel()
188 int bias_; member in tflite::__anon10ad49200111::BaseSVDFOpModel
483 bias_ = AddInput({TensorType_INT32, {units}, -512, 512}); in IntegerSVDFOpModel()
510 QuantizeAndPopulate<int32_t>(bias_, f); in SetBias()
525 int bias_; member in tflite::__anon10ad49200111::IntegerSVDFOpModel
Dtranspose_conv_test.cc601 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseTransposeConvBiasOpModel()
623 bias_ = AddInput(bias); in BaseTransposeConvBiasOpModel()
628 bias_ = AddInput(bias); in BaseTransposeConvBiasOpModel()
640 GetShape(input_), GetShape(bias_)}); in BaseTransposeConvBiasOpModel()
660 QuantizeAndPopulate<int32_t>(bias_, bias); in SetBias()
662 PerChannelQuantizeBias(bias_, bias); in SetBias()
664 PopulateTensor(bias_, bias); in SetBias()
674 int bias_; member in tflite::__anona9e836210111::BaseTransposeConvBiasOpModel
Dconv3d_test.cc41 bias_ = AddInput(bias); in Conv3dOpModel()
49 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in Conv3dOpModel()
73 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias()
83 int bias_; member in tflite::__anonfc97239a0111::Conv3dOpModel
Dconv3d_transpose_test.cc53 bias_ = AddInput(bias); in Conv3dTransposeOpModel()
62 GetShape(input_), GetShape(bias_)}); in Conv3dTransposeOpModel()
100 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias()
111 int bias_; member in tflite::__anonf50558580111::Conv3dTransposeOpModel
Dconv_test.cc71 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
103 bias_ = AddInput(bias);
108 bias_ = AddInput(bias);
122 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)},
130 int bias_; member in tflite::__anoncc1f39a30111::BaseConvolutionOpModel
140 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias()
904 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
1162 PopulateTensor(bias_, data); in SetBias()
1636 PerChannelQuantizeBias(bias_, data); in SetBias()
1776 PopulateTensor(bias_, data); in SetBias()
/external/webrtc/modules/audio_processing/agc2/rnn_vad/
Drnn_fc.cc76 bias_(GetScaledParams(bias)), in FullyConnectedLayer()
82 RTC_DCHECK_EQ(output_size_, bias_.size()) in FullyConnectedLayer()
97 bias_[o] + vector_math_.DotProduct( in ComputeOutput()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc85 bias_ = AddInput(bias);
99 BuildInterpreter({GetShape(input_), GetShape(weights_), GetShape(bias_)},
105 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias()
124 int bias_; member in tflite::__anon59489a3b0111::HybridFullyConnectedOpModel
252 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
284 bias_ = AddInput(bias);
289 bias_ = AddInput(bias);
303 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)},
313 int bias_; member in tflite::__anon59489a3b0111::BaseConvolutionOpModel
330 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
[all …]
/external/libtextclassifier/abseil-cpp/absl/base/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/angle/third_party/abseil-cpp/absl/profiling/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/tensorflow/third_party/absl/abseil-cpp/absl/profiling/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/cronet/third_party/abseil-cpp/absl/profiling/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/base/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/webrtc/third_party/abseil-cpp/absl/profiling/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/abseil-cpp/absl/profiling/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/openscreen/third_party/abseil/src/absl/base/internal/
Dexponential_biased.cc56 double interval = bias_ + (std::log2(q) - 26) * (-std::log(2.0) * mean); in GetSkipCount()
68 bias_ = interval - value; in GetSkipCount()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc84 OP_REQUIRES_OK(context, context->GetAttr("bias", &bias_)); in MklLRNOp()
150 src_dnn_data.GetUsrMemDesc(), kernel_size, new_alpha, beta_, bias_); in Compute()
250 auto tmp = in_shaped.square().contract(multiplier, dims) * alpha_ + bias_; in MklDefaultToEigen()
307 float bias_; member in tensorflow::MklLRNOp
327 OP_REQUIRES_OK(context, context->GetAttr("bias", &bias_)); in MklLRNGradOp()
402 orig_input_md, kernel_size, new_alpha, beta_, bias_); in Compute()
407 target_diff_dst_md, kernel_size, new_alpha, beta_, bias_); in Compute()
572 norm = alpha_ * norm + bias_; in MklDefaultToEigen()
654 float bias_; member in tensorflow::MklLRNGradOp

123