Home
last modified time | relevance | path

Searched refs:filter_dims (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution() argument
50 filter_dims[3]); // filter_count in SpatialConvolution()
55 static_cast<Scalar*>(device_.allocate(BufferSize(filter_dims))); in SpatialConvolution()
60 device_.memset(filter_data, 123, BufferSize(filter_dims)); in SpatialConvolution()
63 Filter filter(filter_data, filter_dims); in SpatialConvolution()
77 Dimensions filter_dims) { in SpatialConvolutionBackwardInput() argument
84 filter_dims[3]); // filter_count in SpatialConvolutionBackwardInput()
91 static_cast<Scalar*>(device_.allocate(BufferSize(filter_dims))); in SpatialConvolutionBackwardInput()
97 device_.memset(filter_data, 123, BufferSize(filter_dims)); in SpatialConvolutionBackwardInput()
100 Filter filter(filter_data, filter_dims); in SpatialConvolutionBackwardInput()
[all …]
Deigen_benchmark_cpu_test.cc44 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution() local
47 benchmark.SpatialConvolution(input_dims, filter_dims); in SpatialConvolution()
72 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput() local
75 benchmark.SpatialConvolutionBackwardInput(input_dims, filter_dims); in SpatialConvolutionBackwardInput()
99 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel() local
102 benchmark.SpatialConvolutionBackwardKernel(input_dims, filter_dims); in SpatialConvolutionBackwardKernel()
104 auto num_computed_elements = filter_dims.TotalSize(); in SpatialConvolutionBackwardKernel()
262 typename Benchmark::Dimensions filter_dims( in CuboidConvolution() local
265 benchmark.CuboidConvolution(input_dims, filter_dims); in CuboidConvolution()
291 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardInput() local
[all …]
Dconv_grad_filter_ops_benchmark_test.cc76 Node* filter_dims = in Conv2DBackpropFilter() local
84 .Input(filter_dims) in Conv2DBackpropFilter()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc246 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; in XLA_TEST_F() local
248 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims); in XLA_TEST_F()
282 auto filter_r5 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in XLA_TEST_F()
312 std::vector<int64> filter_dims = {3, 3, 5, 3}; in RunTest() local
314 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest()
345 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest()
371 std::vector<int64> filter_dims = {3, 3, 1, 15}; in RunTest() local
373 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest()
405 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest()
437 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
[all …]
Dconvolution_test_1d.cc84 std::vector<int64> filter_dims = {window_size, input_feature, in TestImpl() local
87 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in TestImpl()
116 auto filter_r3 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in TestImpl()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc70 cmsis_nn_dims filter_dims; in Prepare() local
71 filter_dims.n = output_shape.Dims(3); in Prepare()
72 filter_dims.h = filter->dims->data[1]; in Prepare()
73 filter_dims.w = filter->dims->data[2]; in Prepare()
74 filter_dims.c = input_dims.c; in Prepare()
97 context, node, params, input_dims.w, input_dims.h, filter_dims.w, in Prepare()
98 filter_dims.h, output_dims.w, output_dims.h, input->type, in Prepare()
116 &conv_params, &input_dims, &filter_dims, &output_dims); in Prepare()
183 cmsis_nn_dims filter_dims; in EvalQuantizedPerChannel() local
184 filter_dims.n = output_depth; in EvalQuantizedPerChannel()
[all …]
Dfully_connected.cc80 cmsis_nn_dims filter_dims; in Prepare() local
81 filter_dims.n = filter_shape.Dims(filter_dim_count - 1); in Prepare()
82 filter_dims.h = 1; in Prepare()
83 filter_dims.w = 1; in Prepare()
84 filter_dims.c = output_shape.Dims(1); in Prepare()
87 arm_fully_connected_s8_get_buffer_size(&filter_dims); in Prepare()
134 cmsis_nn_dims filter_dims; in EvalQuantizedInt8() local
135 filter_dims.n = accum_depth; in EvalQuantizedInt8()
136 filter_dims.h = 1; in EvalQuantizedInt8()
137 filter_dims.w = 1; in EvalQuantizedInt8()
[all …]
Dpooling.cc146 cmsis_nn_dims filter_dims; in AverageEvalQuantized() local
147 filter_dims.n = 1; in AverageEvalQuantized()
148 filter_dims.h = params->filter_height; in AverageEvalQuantized()
149 filter_dims.w = params->filter_width; in AverageEvalQuantized()
150 filter_dims.c = 1; in AverageEvalQuantized()
162 &filter_dims, &output_dims, in AverageEvalQuantized()
236 cmsis_nn_dims filter_dims; in MaxEvalInt8() local
237 filter_dims.n = 1; in MaxEvalInt8()
238 filter_dims.h = params->filter_height; in MaxEvalInt8()
239 filter_dims.w = params->filter_width; in MaxEvalInt8()
[all …]
Ddepthwise_conv.cc182 cmsis_nn_dims filter_dims; in Prepare() local
183 filter_dims.n = 1; in Prepare()
184 filter_dims.h = filter_height; in Prepare()
185 filter_dims.w = filter_width; in Prepare()
186 filter_dims.c = output_depth; in Prepare()
199 &dw_conv_params, &input_dims, &filter_dims, &output_dims); in Prepare()
291 cmsis_nn_dims filter_dims; in EvalQuantizedPerChannel() local
292 filter_dims.n = filter_shape.Dims(0); in EvalQuantizedPerChannel()
293 filter_dims.h = filter_shape.Dims(1); in EvalQuantizedPerChannel()
294 filter_dims.w = filter_shape.Dims(2); in EvalQuantizedPerChannel()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose_conv_test.cc106 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestTransposeConvFloat() local
120 CreateTensor(filter_data, filter_dims), in TestTransposeConvFloat()
142 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestTransposeConvQuantized() local
151 filter_data, filter_quantized, filter_dims, filter_scales, in TestTransposeConvQuantized()
230 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
247 CreateTensor(tflite::testing::kFilterData, filter_dims), in TF_LITE_MICRO_TEST()
264 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
283 CreateQuantizedTensor(filter_data, filter_dims, in TF_LITE_MICRO_TEST()
Dconv_test_common.cc123 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvFloat() local
132 CreateTensor(filter_data, filter_dims), in TestConvFloat()
152 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvQuantizedPerLayer() local
166 CreateQuantizedTensor(filter_data, filter_quantized, filter_dims, in TestConvQuantizedPerLayer()
195 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvQuantizedPerChannel() local
207 filter_data, filter_data_quantized, filter_dims, filter_scales, in TestConvQuantizedPerChannel()
Ddepthwise_conv_test.cc93 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvFloat() local
103 CreateTensor(filter_data, filter_dims), in TestDepthwiseConvFloat()
122 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvQuantizedPerLayer() local
135 filter_dims, filter_scale, in TestDepthwiseConvQuantizedPerLayer()
174 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvQuantizedPerChannel() local
188 filter_data, filter_data_quantized, filter_dims, filter_scales, in TestDepthwiseConvQuantizedPerChannel()
637 const int filter_dims[] = {4, 1, 2, 2, 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()
705 TfLiteIntArray* filter_dims = tflite::testing::IntArrayFromInts(filter_shape); in TF_LITE_MICRO_TEST() local
[all …]
Dconv_test.cc133 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
144 CreateTensor(tflite::testing::kFilterData, filter_dims), in TF_LITE_MICRO_TEST()
162 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
174 CreateQuantizedTensor(filter_data, filter_dims, in TF_LITE_MICRO_TEST()
450 TfLiteIntArray* filter_dims = in TF_LITE_MICRO_TEST() local
469 tflite::testing::kFilterData, filter_quantized, filter_dims, filter_scale, in TF_LITE_MICRO_TEST()
540 TfLiteIntArray* filter_dims = in TF_LITE_MICRO_TEST() local
555 tflite::testing::kFilterData, filter_quantized, filter_dims, in TF_LITE_MICRO_TEST()
650 TfLiteIntArray* filter_dims = tflite::testing::IntArrayFromInts(filter_shape); in TF_LITE_MICRO_TEST() local
681 filter_values, filter_quantized, filter_dims, filter_scale, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h194 memory::dims* filter_dims, in GetFilterSizeInMklOrder() argument
196 DCHECK(filter_dims); in GetFilterSizeInMklOrder()
241 *filter_dims = mkldnn_sizes; in GetFilterSizeInMklOrder()
249 *filter_dims = mkldnn_sizes; in GetFilterSizeInMklOrder()
278 *filter_dims = mkldnn_sizes; in GetFilterSizeInMklOrder()
288 memory::dims* filter_dims, in GetFilterSizeInMklOrder() argument
290 DCHECK(filter_dims); in GetFilterSizeInMklOrder()
292 GetTfShape(context_, filter_index), filter_dims, in GetFilterSizeInMklOrder()
526 memory::dims* input_dims, memory::dims* filter_dims,
532 DCHECK(filter_dims);
[all …]
Dmkl_avgpooling_op.cc84 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
86 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
114 src_dims, output_dims_mkl_order, filter_dims, strides, padding_left, in Compute()
204 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
205 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
243 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
Dmkl_maxpooling_op.cc126 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
127 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
140 src_dims, output_dims_mkl_order, filter_dims, strides, padding_left, in Compute()
260 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
261 this->PoolParamsToDims(&pool_params, &filter_dims, &strides, in Compute()
298 orig_input_dims_mkl_order, output_dims_mkl_order, filter_dims, in Compute()
Dmkl_pooling_ops_common.h43 memory::dims filter_dims; member
54 memory::dims filter_dims, memory::dims strides, in MklPoolingParams()
61 filter_dims(filter_dims), in MklPoolingParams()
191 key_creator.AddAsKey(fwdParams.filter_dims); in CreateKey()
336 key_creator.AddAsKey(bwdParams.filter_dims); in CreateKey()
512 memory::dims* filter_dims, memory::dims* strides, in PoolParamsToDims() argument
517 *filter_dims = in PoolParamsToDims()
527 *filter_dims = in PoolParamsToDims()
Dmkl_conv_grad_input_ops.cc62 memory::dims filter_dims; member
71 MklConvBwdInputParams(memory::dims diff_src_dims, memory::dims filter_dims, in MklConvBwdInputParams()
77 filter_dims(filter_dims), in MklConvBwdInputParams()
196 context_.filter_md.reset(new memory::desc({convBwdInputDims.filter_dims}, in Setup()
279 key_creator.AddAsKey(convBwdInputDims.filter_dims); in CreateKey()
Dmkl_pooling_ops_common.cc55 *context_.dst_md, fwdParams.strides, fwdParams.filter_dims, in Setup()
151 bwdParams.filter_dims, bwdParams.padding_left, bwdParams.padding_right)); in Setup()
156 *context_.dst_md, bwdParams.strides, bwdParams.filter_dims, in Setup()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h43 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
65 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
70 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
77 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
86 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
93 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
102 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
106 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
113 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv() argument
142 DimsToShape(filter_dims), filter_data, DimsToShape(bias_dims), in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dconv.cc262 int* filter_dims = filter->dims->data; in Eval() local
264 input_dims[3] == 32 && filter_dims[0] == 32 && filter_dims[1] == 1 && in Eval()
265 filter_dims[2] == 1 && filter_dims[3] == 32) { in Eval()
Ddepthwise_conv.cc476 int* filter_dims = filter->dims->data; in Eval() local
478 input_dims[3] == 32 && filter_dims[0] == 1 && filter_dims[1] == 4 && in Eval()
479 filter_dims[2] == 1 && filter_dims[3] == 32) { in Eval()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h139 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
164 DimsToShape(filter_dims), filter_data, in DepthwiseConv()
171 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
178 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
187 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
194 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
203 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv() argument
207 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
282 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv() argument
314 op_params, DimsToShape(input_dims), input_data, DimsToShape(filter_dims), in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h552 const float* filter_data, const Dims<4>& filter_dims,
563 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0);
567 const int filter_height = ArraySize(filter_dims, 2);
568 const int filter_width = ArraySize(filter_dims, 1);
653 filter_data + filter_y * filter_dims.strides[2],
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dconv_slicing_test.cc202 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvQuantizedPerChannel() local
215 filter_data, filter_data_quantized, filter_dims, filter_scales, in TestConvQuantizedPerChannel()
220 int channel_count = filter_dims->data[0]; in TestConvQuantizedPerChannel()

12