Home
last modified time | relevance | path

Searched refs:bias_size (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.cu.cc56 T* __restrict__ output, int32 bias_size) { in BiasNHWCKernel() argument
58 int32 bias_offset = index % bias_size; in BiasNHWCKernel()
66 T* __restrict__ output, int32 bias_size, in BiasNCHWKernel() argument
70 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel()
81 const int32 bias_size = channel; in compute() local
83 const int32 total_count = batch * bias_size * image_size; in compute()
92 output, bias_size)); in compute()
97 output, bias_size, image_size)); in compute()
106 int32 bias_size) { in BiasGradNHWC_Naive() argument
108 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive()
[all …]
Dbias_op.h34 const Eigen::Index bias_size = bias.dimension(0); in operator() local
35 const Eigen::Index rest_size = input.size() / bias_size; in operator()
41 const int bias_size = bias.dimension(0); in operator() local
42 const int rest_size = input.size() / bias_size; in operator()
Dmatmul_op_test.cc198 const int bias_size = n; in VerifyBiasAddTensorsNear() local
199 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsNear()
Dmkl_fused_ops_test.cc118 const int bias_size = filter_count; in VerifyBiasAddTensorsClose() local
119 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsClose()
148 const int bias_size = filter_count; in VerifyFusedTensorsClose() local
149 Tensor bias(dtype, {bias_size}); in VerifyFusedTensorsClose()
Dconv_ops_test.cc788 const int bias_size = filter_count; in VerifyBiasAddTensorsNear() local
789 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsNear()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup.cc130 void DecomposeBiasTensor(const int32_t* biases, int bias_size, in DecomposeBiasTensor() argument
135 input_bias->resize(bias_size); in DecomposeBiasTensor()
136 std::copy(biases, biases + bias_size, input_bias->begin()); in DecomposeBiasTensor()
138 cell_bias->resize(bias_size); in DecomposeBiasTensor()
139 std::copy(biases + bias_size, biases + 2 * bias_size, cell_bias->begin()); in DecomposeBiasTensor()
141 forget_bias->resize(bias_size); in DecomposeBiasTensor()
142 std::copy(biases + 2 * bias_size, biases + 3 * bias_size, in DecomposeBiasTensor()
145 output_bias->resize(bias_size); in DecomposeBiasTensor()
146 std::copy(biases + 3 * bias_size, biases + 4 * bias_size, in DecomposeBiasTensor()
Dquant_lstm_sup.h48 void DecomposeBiasTensor(const int32_t* biases, int bias_size,
Dnnapi_delegate_test.cc508 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel() local
510 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in ConvolutionOpModel()
516 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in ConvolutionOpModel()
862 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel() local
864 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in DepthwiseConvolutionOpModel()
870 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in DepthwiseConvolutionOpModel()
Dnnapi_delegate.cc2772 const auto bias_size = output_dims->data[1]; in Map() local
2781 delegate::nnapi::DecomposeBiasTensor(biases_tensor.data.i32, bias_size, in Map()
2787 ANEURALNETWORKS_TENSOR_INT32, kTfLiteInt32, {bias_size}, input_bias, in Map()
2791 ANEURALNETWORKS_TENSOR_INT32, kTfLiteInt32, {bias_size}, in Map()
2795 ANEURALNETWORKS_TENSOR_INT32, kTfLiteInt32, {bias_size}, cell_bias, in Map()
2799 ANEURALNETWORKS_TENSOR_INT32, kTfLiteInt32, {bias_size}, in Map()
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_hybrid_test.cc55 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
57 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
75 {bias_size}, in BaseDepthwiseConvolutionOpModel()
88 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
176 const int bias_size = c; in RandomTest() local
179 std::vector<float> bias_data(bias_size); in RandomTest()
186 for (int i = 0; i < bias_size; ++i) { in RandomTest()
Ddepthwise_conv_test.cc55 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
57 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
75 {bias_size}, in BaseDepthwiseConvolutionOpModel()
88 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
640 const int bias_size = 2400; in TEST_F() local
643 std::vector<float> bias_data(bias_size); in TEST_F()
650 for (int i = 0; i < bias_size; ++i) { in TEST_F()
Dconv_test.cc54 int bias_size = GetShape(filter_)[0]; in BaseConvolutionOpModel() local
56 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseConvolutionOpModel()
74 {bias_size}, in BaseConvolutionOpModel()
87 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseConvolutionOpModel()
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc547 const int bias_size = 4; in TF_LITE_MICRO_TEST() local
550 int32_t bias_quantized[bias_size]; in TF_LITE_MICRO_TEST()
552 int zero_points[bias_size + 1]; in TF_LITE_MICRO_TEST()
553 float scales[bias_size + 1]; in TF_LITE_MICRO_TEST()
586 const int bias_size = 4; in TF_LITE_MICRO_TEST() local
589 int32_t bias_quantized[bias_size]; in TF_LITE_MICRO_TEST()
591 int zero_points[bias_size + 1]; in TF_LITE_MICRO_TEST()
592 float scales[bias_size + 1]; in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc130 int64 bias_size = bias_neon_dims.sizes[0]; in Compute() local
132 bias_size * sizeof(float), Allocator::kAllocatorAlignment)); in Compute()
133 memset(bias_ptr, 0, bias_size * sizeof(float)); in Compute()
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/tests/
Ddepthwise_conv_test.cc33 int bias_size = GetShape(filter_)[3]; in QuantizedDepthwiseConvolutionOpModel() local
36 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in QuantizedDepthwiseConvolutionOpModel()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h67 inline void BiasAndClamp(float clamp_min, float clamp_max, int bias_size, in BiasAndClamp() argument
75 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in BiasAndClamp()
81 for (; array_ptr != array_end_ptr; array_ptr += bias_size) { in BiasAndClamp()
83 for (; i <= bias_size - 16; i += 16) { in BiasAndClamp()
109 for (; i <= bias_size - 4; i += 4) { in BiasAndClamp()
117 for (; i < bias_size; i++) { in BiasAndClamp()
124 array_offset += bias_size) { in BiasAndClamp()
125 for (int i = 0; i < bias_size; i++) { in BiasAndClamp()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc543 int64_t bias_size = bias_and_slice.first; in matchAndRewrite() local
546 RankedTensorType::get({bias_size}, filter_type.getElementType()); in matchAndRewrite()
552 new_bias_values.resize(bias_size, APFloat(0.0)); in matchAndRewrite()
554 new_bias_values.resize(bias_size, *bias_cst.float_value_begin()); in matchAndRewrite()
555 } else if (bias_cst.getNumElements() == bias_size) { // 1-d bias, copy it in matchAndRewrite()
567 int bias_index = (flatten_index++ / slice_size) % bias_size; in matchAndRewrite()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils.h101 const size_t bias_size,
Dquantization_utils_test.cc395 const size_t bias_size = 2; in TEST_F() local
398 AdjustWeightsForBiasScale(quant_params.get(), bias_data, bias_size, in TEST_F()
441 const size_t bias_size = 1; in TEST_F() local
444 AdjustWeightsForBiasScale(quant_params.get(), bias_data, bias_size, in TEST_F()
Dquantization_utils.cc191 const size_t bias_size, in AdjustWeightsForBiasScale() argument
229 const auto minmax = std::minmax_element(bias_data, bias_data + bias_size); in AdjustWeightsForBiasScale()
Dquantize_model.cc1101 uint64_t bias_size; in EnsureBiasScaleCompatibility() local
1102 TF_LITE_ENSURE_STATUS(utils::NumElements(*bias_tensor, &bias_size)); in EnsureBiasScaleCompatibility()
1106 weight_tensor->quantization.get(), bias_data, bias_size, in EnsureBiasScaleCompatibility()
/external/XNNPACK/bench/
Dconvolution.cc485 const float* kernel, size_t kernel_size, const float* bias, size_t bias_size, in compare_with_convolution_f32_reference_output() argument
513 assert(bias_size == groups * group_output_channels); in compare_with_convolution_f32_reference_output()