Home
last modified time | relevance | path

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

12

/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.cc202 const int bias_size = n; in VerifyBiasAddTensorsNear() local
203 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsNear()
Dconv_ops_test.cc795 const int bias_size = filter_count; in VerifyBiasAddTensorsNear() local
796 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_signed_quantization_test.cc246 int bias_size = GetShape(filter_)[0]; local
248 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
271 {bias_size},
284 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale};
Dnnapi_delegate_test.cc601 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel() local
603 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in ConvolutionOpModel()
609 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in ConvolutionOpModel()
964 const int bias_size = GetShape(filter_)[0]; in PerChannelQuantizedConvolutionWithConstantFilterOpModel() local
972 {bias_size}, in PerChannelQuantizedConvolutionWithConstantFilterOpModel()
1064 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel() local
1066 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in DepthwiseConvolutionOpModel()
1072 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in DepthwiseConvolutionOpModel()
Dnnapi_delegate_device_selection_test.cc716 int bias_size = GetShape(filter_)[0]; local
718 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_hybrid_test.cc62 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
64 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
82 {bias_size}, in BaseDepthwiseConvolutionOpModel()
95 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
183 const int bias_size = c; in RandomTest() local
186 std::vector<float> bias_data(bias_size); in RandomTest()
193 for (int i = 0; i < bias_size; ++i) { in RandomTest()
Ddepthwise_conv_test.cc63 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
65 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseDepthwiseConvolutionOpModel()
87 {bias_size}, in BaseDepthwiseConvolutionOpModel()
100 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
652 const int bias_size = 2400; in TEST_F() local
655 std::vector<float> bias_data(bias_size); in TEST_F()
662 for (int i = 0; i < bias_size; ++i) { in TEST_F()
Dtranspose_conv_test.cc590 int bias_size = GetShape(filter_)[0]; in BaseTransposeConvBiasOpModel() local
592 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); in BaseTransposeConvBiasOpModel()
605 {bias_size}, in BaseTransposeConvBiasOpModel()
618 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseTransposeConvBiasOpModel()
Dconv_test.cc69 int bias_size = GetShape(filter_)[0]; local
71 bias_ = AddInput({TensorType_FLOAT32, {bias_size}});
94 {bias_size},
107 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale};
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc98 const int bias_size = NumElements(&bias_tensor); in ProcessPerChannelQuantizedBias() local
99 if (bias_size != per_channel_quant->num_scale_values) { in ProcessPerChannelQuantizedBias()
106 dequantized_bias.reserve(bias_size); in ProcessPerChannelQuantizedBias()
107 for (int i = 0; i < bias_size; ++i) { in ProcessPerChannelQuantizedBias()
121 for (int i = 0; i < bias_size; ++i) { in ProcessPerChannelQuantizedBias()
126 const std::vector<int> bias_shape = {1, 1, 1, bias_size}; in ProcessPerChannelQuantizedBias()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc40 int bias_size = GetShape(filter_)[0]; in QuantizedTransposeConvOpModel() local
54 {bias_size}, in QuantizedTransposeConvOpModel()
67 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in QuantizedTransposeConvOpModel()
Dconv_test.cc41 int bias_size = GetShape(filter_)[0]; in QuantizedConvolutionOpModel() local
43 bias_size = GetShape(filter_)[3]; in QuantizedConvolutionOpModel()
57 {bias_size}, in QuantizedConvolutionOpModel()
70 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in QuantizedConvolutionOpModel()
/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/kernels/internal/
Dcommon.h75 inline void BiasAndClamp(float clamp_min, float clamp_max, int bias_size, in BiasAndClamp() argument
83 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in BiasAndClamp()
89 for (; array_ptr != array_end_ptr; array_ptr += bias_size) { in BiasAndClamp()
91 for (; i <= bias_size - 16; i += 16) { in BiasAndClamp()
117 for (; i <= bias_size - 4; i += 4) { in BiasAndClamp()
125 for (; i < bias_size; i++) { in BiasAndClamp()
132 array_offset += bias_size) { in BiasAndClamp()
133 for (int i = 0; i < bias_size; i++) { in BiasAndClamp()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc664 int64_t bias_size = bias_and_slice.first; in matchAndRewrite() local
667 RankedTensorType::get({bias_size}, filter_type.getElementType()); in matchAndRewrite()
673 new_bias_values.resize(bias_size, APFloat(0.0)); in matchAndRewrite()
675 new_bias_values.resize(bias_size, *bias_cst.float_value_begin()); in matchAndRewrite()
676 } else if (bias_cst.getNumElements() == bias_size) { // 1-d bias, copy it in matchAndRewrite()
688 int bias_index = (flatten_index++ / slice_size) % bias_size; in matchAndRewrite()
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc647 const int bias_size = 4; in TF_LITE_MICRO_TEST() local
650 int32_t bias_quantized[bias_size]; in TF_LITE_MICRO_TEST()
790 const int bias_size = 4; in TF_LITE_MICRO_TEST() local
793 int32_t bias_quantized[bias_size]; in TF_LITE_MICRO_TEST()
795 int zero_points[bias_size + 1]; in TF_LITE_MICRO_TEST()
796 float scales[bias_size + 1]; in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_ops_test.cc129 const int bias_size = filter_count; in VerifyBiasAddTensorsClose() local
130 Tensor bias(dtype, {bias_size}); in VerifyBiasAddTensorsClose()
147 int filter_size, int filter_count, int bias_size, in VerifyFusedTensorsClose() argument
159 Tensor bias(dtype, {bias_size}); in VerifyFusedTensorsClose()
370 const int bias_size = filter_count; in VerifyFusedConv2D() local
373 filter_count, bias_size, fused_ops, run_default, run_fused, padding); in VerifyFusedConv2D()
666 int bias_size, in VerifyFusedDepthwiseConv2D() argument
691 filter_count, bias_size, fused_ops, run_default, run_fused); in VerifyFusedDepthwiseConv2D()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils.h101 const size_t bias_size,
Dquantization_utils_test.cc396 const size_t bias_size = 2; in TEST_F() local
399 AdjustWeightsForBiasScale(quant_params.get(), bias_data, bias_size, in TEST_F()
442 const size_t bias_size = 1; in TEST_F() local
445 AdjustWeightsForBiasScale(quant_params.get(), bias_data, bias_size, in TEST_F()
Dquantization_utils.cc234 const size_t bias_size, in AdjustWeightsForBiasScale() argument
274 const auto minmax = std::minmax_element(bias_data, bias_data + bias_size); in AdjustWeightsForBiasScale()
Dquantize_model.cc1392 uint64_t bias_size; in EnsureBiasScaleCompatibility() local
1393 TF_LITE_ENSURE_STATUS(utils::NumElements(*bias_tensor, &bias_size)); in EnsureBiasScaleCompatibility()
1397 weight_tensor->quantization.get(), bias_data, bias_size, in EnsureBiasScaleCompatibility()

12