Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc36 void FullyConnectedOpBuilder::SetBias(TfLiteTensor* bias) { bias_ = bias; } in SetBias()
69 if (bias_ != nullptr) { in FillCoreMLBias()
71 if (bias_->type == kTfLiteFloat32) { in FillCoreMLBias()
72 const float* bias_data = GetTensorData<float>(bias_); in FillCoreMLBias()
73 std::copy(bias_data, bias_data + NumElements(bias_), in FillCoreMLBias()
77 } else if (bias_->type == kTfLiteFloat16) { in FillCoreMLBias()
82 ->assign(bias_->data.raw, bias_->bytes); in FillCoreMLBias()
Dconvolution_op_builder.cc37 void ConvolutionOpBuilder::SetBias(TfLiteTensor* bias) { bias_ = bias; } in SetBias()
190 if (bias_ != nullptr) { in FillCoreMLBias()
192 if (bias_->type == kTfLiteFloat32) { in FillCoreMLBias()
193 std::copy(bias_->data.f, bias_->data.f + NumElements(bias_->dims), in FillCoreMLBias()
197 } else if (bias_->type == kTfLiteFloat16) { in FillCoreMLBias()
202 ->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/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc63 bias_ = AddInput(bias); in QuantizedTransposeConvOpModel()
68 bias_ = AddInput(bias); in QuantizedTransposeConvOpModel()
88 QuantizeAndPopulate<int32_t>(bias_, bias); in SetBias()
90 PerChannelQuantizeBias(bias_, bias); in SetBias()
93 auto* bias_tensor = interpreter_->tensor(bias_); in SetBias()
108 int bias_; member in tflite::QuantizedTransposeConvOpModel
Dmatmul_test.cc42 bias_ = AddNullInput(); in FullyConnectedOpModel()
46 bias_ = AddInput(bias); in FullyConnectedOpModel()
66 auto* bias_tensor = interpreter_->tensor(bias_); in FullyConnectedOpModel()
72 QuantizeAndPopulate<int>(bias_, data); in SetBias()
99 int bias_; member in tflite::FullyConnectedOpModel
Dconv_test.cc66 bias_ = AddInput(bias); in QuantizedConvolutionOpModel()
71 bias_ = AddInput(bias); in QuantizedConvolutionOpModel()
95 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in QuantizedConvolutionOpModel()
112 QuantizeAndPopulate<int>(bias_, data); in SetBias()
130 PerChannelQuantizeBias(bias_, data); in SetPerChannelQuantizedBias()
136 int bias_; member in tflite::QuantizedConvolutionOpModel
/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::__anonb6d75cba0111::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::__anonb6d75cba0111::LRNGradOp
/external/webrtc/modules/audio_processing/agc2/rnn_vad/
Drnn.cc277 bias_(GetScaledParams(bias)), in FullyConnectedLayer()
284 RTC_DCHECK_EQ(output_size_, bias_.size()) in FullyConnectedLayer()
301 bias_, weights_, in ComputeOutput()
308 ComputeFullyConnectedLayerOutput(input_size_, output_size_, input, bias_, in ComputeOutput()
313 ComputeFullyConnectedLayerOutput(input_size_, output_size_, input, bias_, in ComputeOutput()
327 bias_(GetPreprocessedGruTensor(bias, output_size)), in GatedRecurrentLayer()
335 RTC_DCHECK_EQ(kNumGruGates * output_size_, bias_.size()) in GatedRecurrentLayer()
362 recurrent_weights_, bias_, state_); in ComputeOutput()
369 recurrent_weights_, bias_, state_); in ComputeOutput()
374 recurrent_weights_, bias_, state_); in ComputeOutput()
Drnn.h62 const std::vector<float> bias_; variable
95 const std::vector<float> bias_; variable
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc161 bias_ = AddNullInput();
163 bias_ = AddInput({TensorType_FLOAT32, {units_}});
171 bias_ = AddInput(bias);
174 bias_ = AddInput(bias);
193 inputs.push_back((bias_ == kTfLiteOptionalTensor) ? std::vector<int>()
194 : GetShape(bias_));
206 int bias_; member in tflite::__anonae46fc4d0111::BaseFullyConnectedOpModel
218 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias()
238 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
241 QuantizeAndPopulate<int64_t>(bias_, data); in SetBias64()
[all …]
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::__anon1ff178320111::BaseDepthwiseConvolutionOpModel
141 PopulateTensor(bias_, data); in SetBias()
152 void SetBias(const std::vector<float>& data) { PopulateTensor(bias_, data); } in SetBias()
Dsvdf_test.cc144 bias_ = AddNullInput(); in BaseSVDFOpModel()
188 int bias_; member in tflite::__anon7126f9820111::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::__anon7126f9820111::IntegerSVDFOpModel
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::__anon623e747c0111::Conv3dOpModel
Dtranspose_conv_test.cc592 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseTransposeConvBiasOpModel()
614 bias_ = AddInput(bias); in BaseTransposeConvBiasOpModel()
619 bias_ = AddInput(bias); in BaseTransposeConvBiasOpModel()
631 GetShape(input_), GetShape(bias_)}); in BaseTransposeConvBiasOpModel()
651 QuantizeAndPopulate<int32_t>(bias_, bias); in SetBias()
653 PerChannelQuantizeBias(bias_, bias); in SetBias()
655 PopulateTensor(bias_, bias); in SetBias()
665 int bias_; member in tflite::__anon7fbb49030111::BaseTransposeConvBiasOpModel
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::__anon2c98ea050111::BaseConvolutionOpModel
140 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias()
877 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
1101 PopulateTensor(bias_, data); in SetBias()
1541 PerChannelQuantizeBias(bias_, data); in SetBias()
1659 PopulateTensor(bias_, data); in SetBias()
Ddepthwise_conv_test.cc65 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
96 bias_ = AddInput(bias); in BaseDepthwiseConvolutionOpModel()
101 bias_ = AddInput(bias); in BaseDepthwiseConvolutionOpModel()
123 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in BaseDepthwiseConvolutionOpModel()
129 int bias_; member in tflite::__anond27ad6f10111::BaseDepthwiseConvolutionOpModel
139 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias()
613 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
616 QuantizeAndPopulate<int32_t>(bias_, data); in SetBias()
1614 PerChannelQuantizeBias(bias_, data); in SetBias()
Dbasic_rnn_test.cc181 bias_ = AddInput(TensorType_FLOAT32); in RNNOpModel()
195 void SetBias(std::initializer_list<float> f) { PopulateTensor(bias_, f); } in SetBias()
223 int bias_; member in tflite::__anon3123f5fe0111::RNNOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_signed_quantization_test.cc81 bias_ = AddInput(bias);
95 BuildInterpreter({GetShape(input_), GetShape(weights_), GetShape(bias_)},
101 void SetBias(const std::vector<float>& f) { PopulateTensor(bias_, f); } in SetBias()
120 int bias_; member in tflite::__anonc4d0941d0111::HybridFullyConnectedOpModel
248 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
280 bias_ = AddInput(bias);
285 bias_ = AddInput(bias);
299 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)},
309 int bias_; member in tflite::__anonc4d0941d0111::BaseConvolutionOpModel
326 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/webrtc/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/angle/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/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/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/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()
249 auto tmp = in_shaped.square().contract(multiplier, dims) * alpha_ + bias_; in MklDefaultToEigen()
306 float bias_; member in tensorflow::MklLRNOp
326 OP_REQUIRES_OK(context, context->GetAttr("bias", &bias_)); in MklLRNGradOp()
401 orig_input_md, kernel_size, new_alpha, beta_, bias_); in Compute()
406 target_diff_dst_md, kernel_size, new_alpha, beta_, bias_); in Compute()
571 norm = alpha_ * norm + bias_; in MklDefaultToEigen()
653 float bias_; member in tensorflow::MklLRNGradOp

123