| /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
161 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
167 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
174 bias_dims, stride_width, stride_height, 1, 1, pad_width, in DepthwiseConv()
183 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
190 bias_dims, stride_width, stride_height, pad_width, pad_height, in DepthwiseConv()
199 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
203 bias_dims, stride, stride, pad_width, pad_height, in DepthwiseConv()
210 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
238 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 41 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
62 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
68 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
75 bias_dims, stride_width, stride_height, 1, 1, pad_width, in DepthwiseConv()
84 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
91 bias_dims, stride_width, stride_height, pad_width, pad_height, in DepthwiseConv()
100 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
104 bias_dims, stride, stride, pad_width, pad_height, in DepthwiseConv()
111 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
139 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | depthwise_conv_test.cc | 39 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestDepthwiseConvFloat() local
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"), in TestDepthwiseConvFloat()
120 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestDepthwiseConvQuantized() local
132 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_min, in TestDepthwiseConvQuantized()
|
| D | fully_connected_test.cc | 39 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestFullyConnectedFloat() local
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"), in TestFullyConnectedFloat()
113 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestFullyConnectedQuantized() local
125 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_min, in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 69 memory::dims bias_dims; member
83 memory::dims bias_dims, memory::dims dst_dims, in MklConvFwdParams()
88 bias_dims(bias_dims), in MklConvFwdParams()
225 if (!convFwdDims.bias_dims.empty()) in Setup()
227 {convFwdDims.bias_dims}, MklDnnType<Tbias>(), memory::format::any)); in Setup()
230 if (!convFwdDims.bias_dims.empty()) { in Setup()
297 if (!convFwdDims.bias_dims.empty()) { in Setup()
299 {{{convFwdDims.bias_dims}, MklDnnType<T>(), memory::format::x}, in Setup()
367 key_creator.AddAsKey(convFwdDims.bias_dims); in CreateKey()
1068 memory::dims bias_dims = {}; in Compute() local
[all …]
|
| D | mkl_conv_ops.h | 293 memory::dims* bias_dims) { in GetBiasSizeInMklOrder() argument
299 *bias_dims = {static_cast<int>(bias.dim_size(0))}; in GetBiasSizeInMklOrder()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 358 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction() argument
364 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
372 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction() argument
377 AddBiasAndEvalActivationFunction(bias_data, bias_dims, array_data, array_dims, in AddBiasAndEvalActivationFunction()
933 const Dims<4>& bias_dims, in FullyConnected() argument
943 DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in FullyConnected()
951 const float* bias_data, const Dims<4>& bias_dims, in FullyConnected() argument
956 bias_dims, output_activation_min, output_activation_max, in FullyConnected()
1211 const int32* bias_data, const Dims<4>& bias_dims, in FullyConnected() argument
1228 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | depthwiseconv_float.h | 555 const float* bias_data, const Dims<4>& bias_dims, int stride,
|