Home
last modified time | relevance | path

Searched defs:filter_height (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark_cpu_test.cc35 int filter_count, int filter_height, int filter_width) { in SpatialConvolution()
63 int filter_count, int filter_height, in SpatialConvolutionBackwardInput()
91 int filter_count, int filter_height, in SpatialConvolutionBackwardKernel()
256 int filter_count, int filter_height, int filter_width, in CuboidConvolution()
286 int filter_count, int filter_height, in CuboidConvolutionBackwardInput()
315 int filter_count, int filter_height, in CuboidConvolutionBackwardKernel()
Ddepthwise_conv_op_gpu.h81 const int filter_height = in DepthwiseConv2dGPUKernelNHWC() local
188 const int filter_height = in DepthwiseConv2dGPUKernelNHWCSmall() local
321 const int filter_height = in DepthwiseConv2dGPUKernelNCHW() local
472 const int filter_height = in DepthwiseConv2dGPUKernelNCHWSmall() local
Dconv_ops_using_gemm.cc90 int input_depth, const T2* filter_data, int filter_height, in operator ()()
213 int input_depth, const T2* filter_data, int filter_height, in operator ()()
Dquantized_conv_ops.cc57 int filter_height, int filter_width, int filter_count, in operator ()()
203 int filter_height, int filter_width, int filter_count, in operator ()()
Dmkl_conv_ops_test.cc66 int filter_height; member
/external/tensorflow/tensorflow/lite/kernels/
Dpadding.h44 int in_width, int filter_height, int filter_width, TfLitePadding padding) { in ComputePaddingHeightWidth()
Dtranspose_conv.cc123 const int filter_height = SizeOfDimension(weights, 1); in ResizeIm2ColTensor() local
201 const int filter_height = SizeOfDimension(weights, 1); in Eval() local
Dpooling_test.cc32 int filter_width, int filter_height, in BasePoolingOpModel()
Dconv.cc152 int filter_height = filter->dims->data[1]; in AllocateTemporaryTensorsIfRequired() local
275 int filter_height = filter->dims->data[1]; in Prepare() local
Ddepthwise_conv.cc130 int filter_height = SizeOfDimension(filter, 1); in Prepare() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc55 int height, int filter_width, int filter_height, in CalculateOpData()
123 const int filter_height = filter_shape.Dims(1); in DepthwiseConvOptimizedForFilterWidthEight() local
344 const int filter_height = GetTensorShape(filter).Dims(1); in EvalQuantized() local
401 int filter_height = SizeOfDimension(filter, 1); in Eval() local
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv.cc52 int height, int filter_width, int filter_height, in CalculateOpData()
170 int filter_height = SizeOfDimension(filter, 1); in Eval() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dflatten_atrous.cc69 const int32 filter_height = filter.dim_size(0); in FlattenAtrousConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_float.h49 const int filter_height = filter_shape.Dims(1); in DepthwiseConv() local
Ddepthwiseconv_uint8.h123 const int filter_height = filter_shape.Dims(1); in Run() local
Dlegacy_reference_ops.h1489 int filter_height, float* output_data, in AveragePool()
1497 int pad_height, int filter_width, int filter_height, in AveragePool()
1518 int pad_height, int filter_width, int filter_height, in AveragePool()
1539 int filter_height, int32 output_activation_min, in AveragePool()
1581 int pad_width, int pad_height, int filter_width, int filter_height, in MaxPool()
1589 int pad_height, int filter_width, int filter_height, in MaxPool()
1609 int filter_width, int filter_height, int32 output_activation_min, in MaxPool()
1629 int pad_width, int pad_height, int filter_width, int filter_height, in MaxPool()
1639 int pad_height, int filter_width, int filter_height, in L2Pool()
1659 int filter_width, int filter_height, float* output_data, in L2Pool()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc491 int filter_width, int filter_height, in TryTestDepthwiseConv()
566 const int filter_height = ExponentialRandomPositiveInt(0.9f, 4, 10); in TryTestOneDepthwiseConv() local
588 const int filter_height = 3; in TryTestOneDepthwiseConv3x3Filter() local
640 const int filter_height = 3; in TryTestOneNeonDot3x3() local
Dtest_util.cc24 int filter_width, int filter_height, int stride, in ComputeConvSizes()
Ddepthwiseconv_float_test.cc76 const int filter_height = ExponentialRandomPositiveInt(0.9f, 4, 10); in TryTestOneDepthwiseConv() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h60 const int filter_height = filter_shape.Dims(1); in ConvPerChannel() local
Ddepthwise_conv.h55 const int filter_height = filter_shape.Dims(1); in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h158 const int filter_height = filter_shape.Dims(1); in Conv() local
Dlegacy_optimized_ops.h1239 int filter_height, float* output_data, in AveragePool()
1247 int pad_height, int filter_width, int filter_height, in AveragePool()
1268 int pad_height, int filter_width, int filter_height, in AveragePool()
1289 int filter_height, int32 output_activation_min, in AveragePool()
1331 int pad_width, int pad_height, int filter_width, int filter_height, in MaxPool()
1339 int pad_height, int filter_width, int filter_height, in MaxPool()
1359 int filter_width, int filter_height, int32 output_activation_min, in MaxPool()
1379 int pad_width, int pad_height, int filter_width, int filter_height, in MaxPool()
1389 int pad_height, int filter_width, int filter_height, in L2Pool()
1409 int filter_width, int filter_height, float* output_data, in L2Pool()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py160 def get2d_conv_output_size(input_height, input_width, filter_height, argument
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc181 int64 filter_height = filter_shape.dim_size(0); in CalculateFlops() local
202 int64 filter_height = filter_shape.dim_size(0); in CalculateFlops() local

12