Home
last modified time | relevance | path

Searched defs:bias_dims (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose_conv_test.cc107 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestTransposeConvFloat() local
143 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestTransposeConvQuantized() local
231 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
265 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
Dconv_test.cc134 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
163 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
452 TfLiteIntArray* bias_dims = in TF_LITE_MICRO_TEST() local
542 TfLiteIntArray* bias_dims = in TF_LITE_MICRO_TEST() local
651 TfLiteIntArray* bias_dims = tflite::testing::IntArrayFromInts(bias_shape); in TF_LITE_MICRO_TEST() local
Dconv_test_common.cc124 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvFloat() local
153 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvQuantizedPerLayer() local
196 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvQuantizedPerChannel() local
Ddepthwise_conv_test.cc94 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvFloat() local
123 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvQuantizedPerLayer() local
175 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvQuantizedPerChannel() local
639 const int bias_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
706 TfLiteIntArray* bias_dims = tflite::testing::IntArrayFromInts(bias_shape); in TF_LITE_MICRO_TEST() local
806 TfLiteIntArray* bias_dims = tflite::testing::IntArrayFromInts(bias_shape); in TF_LITE_MICRO_TEST() local
931 TfLiteIntArray* bias_dims = tflite::testing::IntArrayFromInts(bias_shape); in TF_LITE_MICRO_TEST() local
Dfully_connected_test.cc272 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestFullyConnectedFloat() local
304 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestFullyConnectedQuantized() local
Dsvdf_test.cc562 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_args); in TestSVDF() local
617 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestIntegerSVDF() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h44 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
71 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
87 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
103 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
114 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
149 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
169 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
192 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
208 const float* bias_data, const Dims<4>& bias_dims, in Conv()
234 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
172 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
188 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
204 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
283 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
323 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
343 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
366 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
607 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
620 const Dims<4>& bias_dims, in AddBiasAndEvalActivationFunction()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc190 cmsis_nn_dims bias_dims; in EvalQuantizedPerChannel() local
Dfully_connected.cc140 cmsis_nn_dims bias_dims; in EvalQuantizedInt8() local
Ddepthwise_conv.cc297 cmsis_nn_dims bias_dims; in EvalQuantizedPerChannel() local
Dsvdf.cc226 cmsis_nn_dims bias_dims; in EvalIntegerSVDF() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op_fused.cc110 memory::dims bias_dims = memory::dims({channel}); in Compute() local
Dmkl_qmatmul_op.cc225 memory::dims bias_dims = {static_cast<int>(bias_tensor.dim_size(0))}; in Compute() local
Dmkl_conv_ops.h299 memory::dims* bias_dims) { in GetBiasSizeInMklOrder()
Dmkl_matmul_ops_common.h49 memory::dims bias_dims; member
Dmkl_conv_ops.cc61 memory::dims bias_dims; member
649 memory::dims bias_dims = {}; in Compute() local
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dfully_connected_slicing_test.cc48 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestFullyConnectedQuantized() local
Dconv_slicing_test.cc203 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvQuantizedPerChannel() local
Ddepthwise_conv_slicing_test.cc130 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvQuantizedPerChannel() local