Home
last modified time | relevance | path

Searched defs:filter_dims (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 void SpatialConvolution(Dimensions input_dims, Dimensions filter_dims) { in SpatialConvolution()
77 Dimensions filter_dims) { in SpatialConvolutionBackwardInput()
116 Dimensions filter_dims) { in SpatialConvolutionBackwardKernel()
177 void CuboidConvolution(Dimensions input_dims, Dimensions filter_dims) { in CuboidConvolution()
209 Dimensions filter_dims) { in CuboidConvolutionBackwardInput()
250 Dimensions filter_dims) { in CuboidConvolutionBackwardKernel()
Deigen_benchmark_cpu_test.cc44 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolution() local
72 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardInput() local
99 typename Benchmark::Dimensions filter_dims(filter_height, filter_width, in SpatialConvolutionBackwardKernel() local
262 typename Benchmark::Dimensions filter_dims( in CuboidConvolution() local
291 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardInput() local
319 typename Benchmark::Dimensions filter_dims( in CuboidConvolutionBackwardKernel() local
Dconv_grad_filter_ops_benchmark_test.cc76 Node* filter_dims = in Conv2DBackpropFilter() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc246 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; in XLA_TEST_F() local
312 std::vector<int64> filter_dims = {3, 3, 5, 3}; in RunTest() local
371 std::vector<int64> filter_dims = {3, 3, 1, 15}; in RunTest() local
437 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
507 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
571 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
640 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
709 std::vector<int64> filter_dims = {3, 3, 1, 512}; in RunTest() local
780 std::vector<int64> filter_dims = {3, 3, 1, 5}; in RunTest() local
855 std::vector<int64> filter_dims = {3, 3, 1, 160}; in RunTest() local
[all …]
Dconvolution_test_1d.cc84 std::vector<int64> filter_dims = {window_size, input_feature, in TestImpl() local
/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose_conv_test.cc106 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestTransposeConvFloat() local
142 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestTransposeConvQuantized() local
230 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
264 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
Dconv_test.cc133 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
162 TfLiteIntArray* filter_dims = IntArrayFromInts(tflite::testing::kFilterShape); in TF_LITE_MICRO_TEST() local
450 TfLiteIntArray* filter_dims = in TF_LITE_MICRO_TEST() local
540 TfLiteIntArray* filter_dims = in TF_LITE_MICRO_TEST() local
650 TfLiteIntArray* filter_dims = tflite::testing::IntArrayFromInts(filter_shape); in TF_LITE_MICRO_TEST() local
Dconv_test_common.cc123 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvFloat() local
152 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvQuantizedPerLayer() local
195 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvQuantizedPerChannel() local
Ddepthwise_conv_test.cc93 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvFloat() local
122 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvQuantizedPerLayer() local
174 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvQuantizedPerChannel() local
637 const int filter_dims[] = {4, 1, 2, 2, 4}; in TF_LITE_MICRO_TEST() local
705 TfLiteIntArray* filter_dims = tflite::testing::IntArrayFromInts(filter_shape); in TF_LITE_MICRO_TEST() local
805 TfLiteIntArray* filter_dims = tflite::testing::IntArrayFromInts(filter_shape); in TF_LITE_MICRO_TEST() local
930 TfLiteIntArray* filter_dims = tflite::testing::IntArrayFromInts(filter_shape); in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc70 cmsis_nn_dims filter_dims; in Prepare() local
183 cmsis_nn_dims filter_dims; in EvalQuantizedPerChannel() local
Dfully_connected.cc80 cmsis_nn_dims filter_dims; in Prepare() local
134 cmsis_nn_dims filter_dims; in EvalQuantizedInt8() local
Dpooling.cc146 cmsis_nn_dims filter_dims; in AverageEvalQuantized() local
236 cmsis_nn_dims filter_dims; in MaxEvalInt8() local
Ddepthwise_conv.cc182 cmsis_nn_dims filter_dims; in Prepare() local
291 cmsis_nn_dims filter_dims; in EvalQuantizedPerChannel() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc84 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
204 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
Dmkl_maxpooling_op.cc126 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
260 memory::dims filter_dims, strides, padding_left, padding_right; in Compute() local
Dmkl_conv_ops.h194 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
288 memory::dims* filter_dims, in GetFilterSizeInMklOrder()
Dmkl_pooling_ops_common.h43 memory::dims filter_dims; member
512 memory::dims* filter_dims, memory::dims* strides, in PoolParamsToDims()
Dmkl_conv_grad_input_ops.cc62 memory::dims filter_dims; member
Dmkl_conv_ops.cc60 memory::dims filter_dims; member
514 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h43 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
70 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
86 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
102 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
113 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
148 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
168 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
191 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
207 const float* filter_data, const Dims<4>& filter_dims, in Conv()
233 const float* filter_data, const Dims<4>& filter_dims, in Conv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h139 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
171 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
187 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
203 const float* filter_data, const Dims<4>& filter_dims, in DepthwiseConv()
282 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
322 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
342 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
365 const Dims<4>& filter_dims, int32 filter_offset, in DepthwiseConv()
1723 const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
1748 const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dconv.cc262 int* filter_dims = filter->dims->data; in Eval() local
Ddepthwise_conv.cc476 int* filter_dims = filter->dims->data; in Eval() local
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dconv_slicing_test.cc202 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestConvQuantizedPerChannel() local
Ddepthwise_conv_slicing_test.cc129 TfLiteIntArray* filter_dims = IntArrayFromInts(filter_dims_data); in TestDepthwiseConvQuantizedPerChannel() local

12