/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_fused_ops_test.cc | 43 const Tensor& bias_data, Tensor* out)>; 133 const Tensor& bias_data, Tensor* output, in RunConv2DWithBias() argument 145 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBias() 152 const Tensor& bias_data, Tensor* output, in RunConv2DWithBiasAndRelu() argument 164 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBiasAndRelu() 223 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias() argument 224 RunConv2DWithBias(input_data, filter_data, bias_data, out); in VerifyConv2DWithBias() 229 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias() argument 230 RunMklFusedConv2DOp(input_data, filter_data, {bias_data}, {"BiasAdd"}, in VerifyConv2DWithBias() 248 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu() argument [all …]
|
D | conv_ops_test.cc | 540 const Tensor& bias_data, Tensor* out)>; 610 const Tensor& bias_data, Tensor* output, in RunConv2DWithBias() argument 622 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBias() 629 const Tensor& bias_data, Tensor* output, in RunConv2DWithBiasAndRelu() argument 641 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBiasAndRelu() 842 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias() argument 843 RunConv2DWithBias(input_data, filter_data, bias_data, out); in VerifyConv2DWithBias() 848 const Tensor& bias_data, in VerifyConv2DWithBias() 850 RunFusedConv2DOp(input_data, filter_data, {bias_data}, {"BiasAdd"}, out); in VerifyConv2DWithBias() 867 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu() argument [all …]
|
D | conv_ops_fused_impl.h | 159 explicit BiasAddOutputKernel(const T* bias_data) : bias_data(bias_data) {} in BiasAddOutputKernel() 168 const T* bias_base = bias_data + i; in operator() 180 const T* bias_data;
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | fully_connected.h | 33 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument 56 if (bias_data) { in FullyConnected() 57 bias_value = bias_data[out_c]; in FullyConnected() 69 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument 102 if (bias_data) { in FullyConnected() 103 acc += bias_data[out_c]; in FullyConnected() 118 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument 146 int32 accum = bias_data[out_c]; in FullyConnected() 172 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected() argument 252 int32 acc = accum[i] + bias_data[c + i]; in ShuffledFullyConnected() [all …]
|
D | depthwiseconv_uint8.h | 97 const RuntimeShape& bias_shape, const int32* bias_data, in Run() 158 if (bias_data) { in Run() 159 acc += bias_data[oc]; in Run() 182 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument 188 bias_data, output_shape, in DepthwiseConv()
|
D | legacy_reference_ops.h | 41 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 63 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv() 68 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 74 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 84 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 90 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 100 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument 103 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 111 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 140 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv() [all …]
|
D | depthwiseconv_float.h | 29 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument 83 if (bias_data) { in DepthwiseConv() 84 bias_value = bias_data[oc]; in DepthwiseConv()
|
/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | depthwiseconv_quantized_test.cc | 109 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv() argument 142 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 166 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 200 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 217 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 236 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 245 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 274 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 280 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv() 292 std::int32_t filter_offset, const std::int32_t* bias_data, in TestOneDepthwiseConvWithGivenOutputShift() argument [all …]
|
D | depthwiseconv_float_test.cc | 36 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv() argument 41 filter_data, bias_shape, bias_data, output_shape, in TestOneDepthwiseConv() 44 filter_data, bias_shape, bias_data, output_shape, in TestOneDepthwiseConv() 119 std::vector<float> bias_data(output_depth); in TryTestOneDepthwiseConv() local 126 FillRandom(&bias_data, -bias_amplitude, bias_amplitude); in TryTestOneDepthwiseConv() 140 bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | legacy_optimized_ops.h | 140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 162 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv() 167 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 173 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 183 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 189 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 199 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument 202 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv() 210 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument 239 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv() [all …]
|
D | depthwiseconv_uint8_3x3_filter.h | 3281 const uint8* filter_data, const int32* bias_data, 3306 const int32* bias_ptr = bias_data; 3363 ConvKernel::Run(input_data, filter_data, bias_data, output_data, 0, 3379 const int32* bias_data, 3386 input_data, filter_ptr, bias_data, output_data, ¶ms); 3402 input_ptr, filter_ptr, bias_data, output_ptr, ¶ms); 3411 input_ptr, filter_ptr, bias_data, output_ptr, ¶ms); 3417 input_ptr, filter_ptr, bias_data, output_ptr, ¶ms); 3427 input_ptr, filter_ptr, bias_data, output_ptr, ¶ms); 3442 input_ptr, filter_ptr, bias_data, output_ptr, ¶ms); [all …]
|
D | depthwiseconv_uint8_transitional.h | 114 const int32* bias_data, 124 adjusted_bias_block[s][z] = bias_data[4 * s + z]; 133 static void Run(const uint8* filter_data, const int32* bias_data, 148 bias_data + 2 * bias_increment * j_depth, adjusted_bias_block); 162 static inline void Run(const uint8* filter_data, const int32* bias_data, 247 adjusted_bias_data_0[z] = bias_data[z]; 248 adjusted_bias_data_1[z] = bias_data[4 + z]; 267 bias_data += 2 * bias_increment; 278 const uint8* filter_data, const int32* bias_data, 363 int32x4_t adjusted_bias_data_a = vld1q_s32(bias_data); [all …]
|
D | depthwiseconv_uint8.h | 1589 const int32* bias_data, 1594 const int32x4_t b = vdupq_n_s32(bias_data[0]); 1605 int32x4_t b = vdupq_n_s32(bias_data[0]); 1606 b = vsetq_lane_s32(bias_data[1], b, 1); 1607 b = vsetq_lane_s32(bias_data[1], b, 3); 1618 const int32x4_t b = vld1q_s32(bias_data); 1629 const int32x4_t b0 = vld1q_s32(bias_data); 1630 const int32x4_t b1 = vld1q_s32(bias_data + 4); 1642 const int32x4_t b0 = vld1q_s32(bias_data); 1643 const int32x4_t b1 = vld1q_s32(bias_data + 4); [all …]
|
D | multithreaded_conv.h | 139 const float* bias_data, const RuntimeShape& output_shape, in Conv() argument 170 output_activation_min, output_activation_max, bias_shape, bias_data, in Conv()
|
D | optimized_ops.h | 190 const float* bias_data, in AddBiasAndEvalActivationFunction() argument 205 auto b0 = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction() 206 auto b1 = vld1q_f32(bias_data + i + 4); in AddBiasAndEvalActivationFunction() 207 auto b2 = vld1q_f32(bias_data + i + 8); in AddBiasAndEvalActivationFunction() 208 auto b3 = vld1q_f32(bias_data + i + 12); in AddBiasAndEvalActivationFunction() 231 auto b = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction() 239 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i], in AddBiasAndEvalActivationFunction() 253 array_data[array_offset + i] + bias_data[i], output_activation_min, in AddBiasAndEvalActivationFunction() 298 const int32* bias_data, int32 accum_multiplier, in GEMVForLstmCell() argument 318 const int32* bias_ptr = bias_data; in GEMVForLstmCell() [all …]
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
D | conv.h | 29 const int32* bias_data, const RuntimeShape& output_shape, in ConvPerChannel() argument 53 if (bias_data) { in ConvPerChannel() 109 if (bias_data) { in ConvPerChannel() 110 acc += bias_data[out_channel]; in ConvPerChannel()
|
D | fully_connected.h | 27 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument 54 if (bias_data) { in FullyConnected() 55 acc += bias_data[out_c]; in FullyConnected()
|
D | depthwise_conv.h | 27 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel() argument 105 if (bias_data) { in DepthwiseConvPerChannel() 106 acc += bias_data[output_channel]; in DepthwiseConvPerChannel()
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | fuse_binary_into_preceding_affine.cc | 47 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine() local 75 float& bias_val = bias_data[i]; in FuseAddOrSubParamsIntoPrecedingAffine() 118 float* const bias_data = bias_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local 154 bias_data[c] *= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine() 156 bias_data[c] /= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
|
D | fuse_binary_into_following_affine.cc | 75 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() local 92 bias_data[d] += accumulation; in FuseAddOrSubParamsIntoFollowingAffine() 106 bias_data[c] += accumulation; in FuseAddOrSubParamsIntoFollowingAffine()
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 285 const float* bias_data, in AddBiasAndEvalActivationFunction() argument 300 auto b0 = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction() 301 auto b1 = vld1q_f32(bias_data + i + 4); in AddBiasAndEvalActivationFunction() 302 auto b2 = vld1q_f32(bias_data + i + 8); in AddBiasAndEvalActivationFunction() 303 auto b3 = vld1q_f32(bias_data + i + 12); in AddBiasAndEvalActivationFunction() 326 auto b = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction() 334 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i], in AddBiasAndEvalActivationFunction() 348 array_data[array_offset + i] + bias_data[i], output_activation_min, in AddBiasAndEvalActivationFunction() 357 inline void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction() argument 364 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction() [all …]
|
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
D | depthwise_conv_test.cc | 32 std::initializer_list<float> bias_data, in TestDepthwiseConvFloat() argument 49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"), in TestDepthwiseConvFloat() 114 std::initializer_list<int32_t> bias_data, float bias_min, float bias_max, in TestDepthwiseConvQuantized() argument 132 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_min, in TestDepthwiseConvQuantized()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | fully_connected_test.cc | 604 std::vector<float> bias_data(output_depth); in SimpleTestQuantizedInt16OutputCase() local 608 for (auto& b : bias_data) { in SimpleTestQuantizedInt16OutputCase() 612 m.SetBias(bias_data); in SimpleTestQuantizedInt16OutputCase() 620 float accum = bias_data[o]; in SimpleTestQuantizedInt16OutputCase()
|
/external/tensorflow/tensorflow/core/kernels/neon/ |
D | depthwiseconv_float.h | 541 const float* bias_data, 547 memcpy(acc_buffer + i * output_depth, bias_data, 555 const float* bias_data, const Dims<4>& bias_dims, int stride, 645 DepthwiseConvInitAccBuffer(num_output_pixels, output_depth, bias_data,
|
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/ |
D | depthwise_conv.cc | 99 const RuntimeShape& bias_shape, const int32* bias_data, in DepthwiseConvOptimizedForFilterWidthEight() argument 255 if (bias_data) { in DepthwiseConvOptimizedForFilterWidthEight() 256 acc += bias_data[oc]; in DepthwiseConvOptimizedForFilterWidthEight()
|