Searched defs:bias_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 140 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()
303 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
316 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
[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()
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 …]
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | depthwise_conv_test.cc | 39 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestDepthwiseConvFloat() local
120 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestDepthwiseConvQuantized() local
|
| D | fully_connected_test.cc | 39 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestFullyConnectedFloat() local
113 TfLiteIntArray* bias_dims = IntArrayFromInitializer(bias_dims_data); in TestFullyConnectedQuantized() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 358 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
372 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
933 const Dims<4>& bias_dims, in FullyConnected()
951 const float* bias_data, const Dims<4>& bias_dims, in FullyConnected()
1211 const int32* bias_data, const Dims<4>& bias_dims, in FullyConnected()
1325 const int32* bias_data_int32, const Dims<4>& bias_dims, int32 output_offset, in FullyConnected()
1351 const int32* bias_data, const Dims<4>& bias_dims, in FullyConnected()
1817 const int32* bias_data, const Dims<4>& bias_dims, int32 output_multiplier, in ShuffledFullyConnected()
2178 const float* bias_data, const Dims<4>& bias_dims, in Conv()
2288 const float* bias_data, const Dims<4>& bias_dims, in HybridConv()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 293 memory::dims* bias_dims) { in GetBiasSizeInMklOrder()
|
| D | mkl_conv_ops.cc | 69 memory::dims bias_dims; member
1068 memory::dims bias_dims = {}; in Compute() local
|