Home
last modified time | relevance | path

Searched defs:bias_data (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_ops_test.cc133 const Tensor& bias_data, Tensor* output, in RunConv2DWithBias()
152 const Tensor& bias_data, Tensor* output, in RunConv2DWithBiasAndRelu()
223 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias()
229 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias()
248 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu()
254 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu()
502 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBias()
520 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
527 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
539 const Tensor& bias_data, Tensor* output, in RunMklPadWithFusedConv2DAndBias()
[all …]
Dconv_ops_test.cc610 const Tensor& bias_data, Tensor* output, in RunConv2DWithBias()
629 const Tensor& bias_data, Tensor* output, in RunConv2DWithBiasAndRelu()
842 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias()
867 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu()
874 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu()
Dmkl_conv_ops.cc121 const Tbias* bias_data, const Toutput* dst_data) { in Execute()
1148 Tbias* bias_data = in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h33 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
69 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
118 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
172 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
Dlegacy_reference_ops.h41 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
68 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
84 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
100 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
111 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
146 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
166 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
189 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
205 const float* bias_data, const Dims<4>& bias_dims, in Conv()
231 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
[all …]
Ddepthwiseconv_float.h29 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConv()
Ddepthwiseconv_uint8.h182 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConv()
Dreference_ops.h136 const float* bias_data, const RuntimeShape& output_shape, in Conv()
209 const int32* bias_data, const RuntimeShape& output_shape, in Conv()
1840 const float* bias_data, const RuntimeShape& unextended_prev_state_shape, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_float_test.cc36 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv()
119 std::vector<float> bias_data(output_depth); in TryTestOneDepthwiseConv() local
Ddepthwiseconv_quantized_test.cc109 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
292 std::int32_t filter_offset, const std::int32_t* bias_data, in TestOneDepthwiseConvWithGivenOutputShift()
425 std::int32_t filter_offset, const std::int32_t* bias_data, in TestOneDepthwiseConvBisectOutputShift()
475 std::int32_t filter_offset, const std::int32_t* bias_data, in TestOneDepthwiseConv()
537 std::vector<std::int32_t> bias_data(output_depth); in TryTestDepthwiseConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
167 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
183 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
199 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
210 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
245 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
265 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
288 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
302 inline void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction()
315 void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction()
[all …]
Dmultithreaded_conv.h139 const float* bias_data, const RuntimeShape& output_shape, in Conv()
Doptimized_ops.h190 const float* bias_data, in AddBiasAndEvalActivationFunction()
298 const int32* bias_data, int32 accum_multiplier, in GEMVForLstmCell()
485 const RuntimeShape& bias_shape, const int32* bias_data, in GEMVForLstmCellWithSymmetricRange()
820 const RuntimeShape& bias_shape, const int32* bias_data, int32 output_offset, in FullyConnectedAsGEMVWorkerImpl()
1096 const RuntimeShape& bias_shape, const int32* bias_data, int32 output_offset, in FullyConnectedAsGEMV()
1170 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
1320 int output_stride, int accum_depth, const int32* bias_data, in ShuffledFullyConnectedWorkerImpl()
1635 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
2183 const float* bias_data, const RuntimeShape& output_shape, in Conv()
2288 const RuntimeShape& bias_shape, const float* bias_data, in HybridConv()
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc47 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine() local
118 float* const bias_data = bias_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
Dfuse_binary_into_following_affine.cc75 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h27 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
Dconv.h29 const int32* bias_data, const RuntimeShape& output_shape, in ConvPerChannel()
Ddepthwise_conv.h27 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv_test.cc32 std::initializer_list<float> bias_data, in TestDepthwiseConvFloat()
114 std::initializer_list<int32_t> bias_data, float bias_min, float bias_max, in TestDepthwiseConvQuantized()
Dfully_connected_test.cc32 std::initializer_list<float> bias_data, in TestFullyConnectedFloat()
107 std::initializer_list<int32_t> bias_data, float bias_min, float bias_max, in TestFullyConnectedQuantized()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h285 const float* bias_data, in AddBiasAndEvalActivationFunction()
357 inline void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction()
371 void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction()
420 const int32* bias_data, int32 accum_multiplier, in GEMVForLstmCell()
607 const RuntimeShape& bias_shape, const int32* bias_data, in GEMVForLstmCellWithSymmetricRange()
899 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
932 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
951 const float* bias_data, const Dims<4>& bias_dims, in FullyConnected()
965 const RuntimeShape& bias_shape, const int32* bias_data, int32 output_offset, in FullyConnectedAsGEMV()
1153 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc442 const float* bias_data = in ConvertConvOperator() local
532 const float* bias_data = in ConvertDepthwiseConvOperator() local
666 const float* bias_data = in ConvertFullyConnectedOperator() local
1515 const float* bias_data = in ConvertLstmCellOperator() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc99 const RuntimeShape& bias_shape, const int32* bias_data, in DepthwiseConvOptimizedForFilterWidthEight()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc604 std::vector<float> bias_data(output_depth); in SimpleTestQuantizedInt16OutputCase() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1180 miopenStatus_t SetBiasArgs(const void* bias_data) { in SetBiasArgs()
4029 const DeviceMemory<T>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivationImpl()
4117 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation()