Home
last modified time | relevance | path

Searched refs:bias_dims (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose_conv_test.cc107 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestTransposeConvFloat() local
122 CreateTensor(bias_data, bias_dims), in TestTransposeConvFloat()
143 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestTransposeConvQuantized() local
167 CreateQuantizedBiasTensor(bias_data, bias_quantized, bias_dims, in TestTransposeConvQuantized()
231 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
248 CreateTensor(tflite::testing::kBiasData, bias_dims), in TF_LITE_MICRO_TEST()
265 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
286 CreateTensor(tflite::testing::kBiasData, bias_dims), in TF_LITE_MICRO_TEST()
Dconv_test_common.cc124 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvFloat() local
133 CreateTensor(bias_data, bias_dims), in TestConvFloat()
153 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvQuantizedPerLayer() local
168 CreateQuantizedBiasTensor(bias_data, bias_quantized, bias_dims, in TestConvQuantizedPerLayer()
196 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvQuantizedPerChannel() local
210 bias_data, bias_data_quantized, bias_dims, input_scale, &filter_scales[1], in TestConvQuantizedPerChannel()
Ddepthwise_conv_test.cc94 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvFloat() local
104 CreateTensor(bias_data, bias_dims), in TestDepthwiseConvFloat()
123 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvQuantizedPerLayer() local
138 bias_data, bias_quantized, bias_dims, input_scale, filter_scale), in TestDepthwiseConvQuantizedPerLayer()
175 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvQuantizedPerChannel() local
192 bias_data, bias_data_quantized, bias_dims, input_scale, &filter_scales[1], in TestDepthwiseConvQuantizedPerChannel()
639 const int bias_dims[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
667 filter_dims, filter_data, filter_quantized, bias_dims, bias_data, in TF_LITE_MICRO_TEST()
672 input_dims, input_data, filter_dims, filter_data, bias_dims, bias_data, in TF_LITE_MICRO_TEST()
706 TfLiteIntArray* bias_dims = tflite::testing::IntArrayFromInts(bias_shape); in TF_LITE_MICRO_TEST() local
[all …]
Dconv_test.cc134 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
145 CreateTensor(tflite::testing::kBiasData, bias_dims), in TF_LITE_MICRO_TEST()
163 TfLiteIntArray* bias_dims = IntArrayFromInts(tflite::testing::kBiasShape); in TF_LITE_MICRO_TEST() local
177 CreateTensor(tflite::testing::kBiasData, bias_dims), in TF_LITE_MICRO_TEST()
452 TfLiteIntArray* bias_dims = in TF_LITE_MICRO_TEST() local
483 tflite::testing::CreateTensor(bias_quantized, bias_dims); in TF_LITE_MICRO_TEST()
542 TfLiteIntArray* bias_dims = in TF_LITE_MICRO_TEST() local
560 tflite::testing::kBiasData, bias_quantized, bias_dims, input_scale, in TF_LITE_MICRO_TEST()
651 TfLiteIntArray* bias_dims = tflite::testing::IntArrayFromInts(bias_shape); in TF_LITE_MICRO_TEST() local
696 tflite::testing::CreateTensor(bias_quantized, bias_dims); in TF_LITE_MICRO_TEST()
Dsvdf_test.cc562 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_args); in TestSVDF() local
572 CreateTensor(bias_data, bias_dims), in TestSVDF()
617 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestIntegerSVDF() local
636 CreateQuantizedBiasTensor(bias_data, bias_quantized, bias_dims, in TestIntegerSVDF()
Dfully_connected_test.cc272 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestFullyConnectedFloat() local
282 CreateTensor(bias_data, bias_dims), in TestFullyConnectedFloat()
304 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestFullyConnectedQuantized() local
316 CreateQuantizedBiasTensor(bias_data, bias_quantized, bias_dims, in TestFullyConnectedQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h44 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
65 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
71 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
78 bias_dims, stride_width, stride_height, 1, 1, pad_width, in DepthwiseConv()
87 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
94 bias_dims, stride_width, stride_height, pad_width, pad_height, in DepthwiseConv()
103 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
107 bias_dims, stride, stride, pad_width, pad_height, in DepthwiseConv()
114 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
142 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dfully_connected.cc140 cmsis_nn_dims bias_dims; in EvalQuantizedInt8() local
141 bias_dims.n = 1; in EvalQuantizedInt8()
142 bias_dims.h = 1; in EvalQuantizedInt8()
143 bias_dims.w = 1; in EvalQuantizedInt8()
144 bias_dims.c = output_depth; in EvalQuantizedInt8()
165 tflite::micro::GetTensorData<int8_t>(filter), &bias_dims, in EvalQuantizedInt8()
Dconv.cc190 cmsis_nn_dims bias_dims; in EvalQuantizedPerChannel() local
191 bias_dims.n = 1; in EvalQuantizedPerChannel()
192 bias_dims.h = 1; in EvalQuantizedPerChannel()
193 bias_dims.w = 1; in EvalQuantizedPerChannel()
194 bias_dims.c = output_depth; in EvalQuantizedPerChannel()
221 tflite::micro::GetTensorData<int8_t>(filter), &bias_dims, in EvalQuantizedPerChannel()
Ddepthwise_conv.cc297 cmsis_nn_dims bias_dims; in EvalQuantizedPerChannel() local
298 bias_dims.n = 1; in EvalQuantizedPerChannel()
299 bias_dims.h = 1; in EvalQuantizedPerChannel()
300 bias_dims.w = 1; in EvalQuantizedPerChannel()
301 bias_dims.c = output_depth; in EvalQuantizedPerChannel()
322 tflite::micro::GetTensorData<int8_t>(filter), &bias_dims, in EvalQuantizedPerChannel()
Dsvdf.cc226 cmsis_nn_dims bias_dims; in EvalIntegerSVDF() local
227 bias_dims.n = bias_tensor->dims->data[0]; in EvalIntegerSVDF()
277 &bias_dims, (int32_t*)tflite::micro::GetTensorData<int32_t>(bias_tensor), in EvalIntegerSVDF()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
165 DimsToShape(bias_dims), bias_data, output_shape, in DepthwiseConv()
172 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
179 bias_dims, stride_width, stride_height, 1, 1, pad_width, in DepthwiseConv()
188 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
195 bias_dims, stride_width, stride_height, pad_width, pad_height, in DepthwiseConv()
204 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
208 bias_dims, stride, stride, pad_width, pad_height, in DepthwiseConv()
283 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
315 filter_data, DimsToShape(bias_dims), bias_data, DimsToShape(output_dims), in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_ops_common.h49 memory::dims bias_dims; member
62 memory::dims bias_dims, memory::dims dst_dims,
68 bias_dims(bias_dims), in src_dims()
191 context_.bias_md.reset(new memory::desc({matmul_fwd_params.bias_dims}, in Setup()
271 context_.bias_mem.reset(new memory({{matmul_fwd_params.bias_dims}, in Setup()
338 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.bias_dims); in CreateKey()
Dmkl_conv_ops.cc61 memory::dims bias_dims; member
79 memory::dims bias_dims, memory::dims dst_dims, in MklConvFwdParams()
85 bias_dims(bias_dims), in MklConvFwdParams()
227 if (!convFwdDims.bias_dims.empty()) in Setup()
228 context_.bias_md.reset(new memory::desc({convFwdDims.bias_dims}, in Setup()
233 if (!convFwdDims.bias_dims.empty()) { in Setup()
294 if (!convFwdDims.bias_dims.empty()) { in Setup()
296 {{convFwdDims.bias_dims}, MklDnnType<Tbias>(), memory::format_tag::x}, in Setup()
367 key_creator.AddAsKey(convFwdDims.bias_dims); in CreateKey()
649 memory::dims bias_dims = {}; in Compute() local
[all …]
Dmkl_matmul_op_fused.cc110 memory::dims bias_dims = memory::dims({channel}); in Compute() local
120 src_dims, weight_dims, bias_dims, dst_dims, src_format, in Compute()
Dmkl_qmatmul_op.cc225 memory::dims bias_dims = {static_cast<int>(bias_tensor.dim_size(0))}; in Compute() local
227 MklDnnMatMulFwdParams matmul_fwd_dims(src_dims, weight_dims, bias_dims, in Compute()
Dmkl_conv_ops.h299 memory::dims* bias_dims) { in GetBiasSizeInMklOrder() argument
305 *bias_dims = {static_cast<int>(bias.dim_size(0))}; in GetBiasSizeInMklOrder()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dfully_connected_slicing_test.cc48 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestFullyConnectedQuantized() local
60 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_scale), in TestFullyConnectedQuantized()
Ddepthwise_conv_slicing_test.cc130 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestDepthwiseConvQuantizedPerChannel() local
148 bias_data, bias_data_quantized, bias_dims, input_scale, &filter_scales[1], in TestDepthwiseConvQuantizedPerChannel()
Dconv_slicing_test.cc203 TfLiteIntArray* bias_dims = IntArrayFromInts(bias_dims_data); in TestConvQuantizedPerChannel() local
233 bias_data, bias_data_quantized, bias_dims, input_scale, &filter_scales[1], in TestConvQuantizedPerChannel()
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h553 const float* bias_data, const Dims<4>& bias_dims, int stride,