Home
last modified time | relevance | path

Searched defs:input_rows (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc32 const int input_rows = 3; in TEST() local
81 const int input_rows = 3; in TEST() local
133 const int input_rows = 7; in TEST() local
181 const int input_rows = 7; in TEST() local
233 const int input_rows = 7; in TEST() local
288 const int input_rows = 7; in TEST() local
341 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid()
400 int input_rows = 3; in TEST() local
417 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
477 int input_rows = 3; in TEST() local
[all …]
Deigen_pooling_test.cc29 const int input_rows = 5; in TEST() local
77 const int input_rows = 5; in TEST() local
127 const int input_rows = 5; in TEST() local
187 const int input_rows = 5; in TEST() local
247 const int input_rows = 5; in TEST() local
309 const int input_rows = 5; in TEST() local
371 const int input_rows = 5; in TEST() local
449 const int input_rows = 5; in TEST() local
526 const int input_rows = 5; in TEST() local
574 const int input_rows = 5; in TEST() local
[all …]
Ddilation_ops_gpu.cu.cc40 const T* __restrict__ filter_ptr, int batch, int input_rows, int input_cols, in DilationKernel()
80 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropInputKernel()
130 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropFilterKernel()
186 const int input_rows = input.dimension(1); in operator ()() local
217 const int input_rows = input.dimension(1); in operator ()() local
259 const int input_rows = input.dimension(1); in operator ()() local
Deigen_backward_cuboid_convolutions_test.cc33 const int input_rows = 3; in TEST() local
96 const int input_rows = 3; in TEST() local
159 const int input_rows = 3; in TEST() local
230 const int input_rows = 3; in TEST() local
302 const int input_rows = 3; in TEST() local
369 const int input_rows = 3; in TEST() local
435 const int input_rows = 3; in TEST() local
501 const int input_rows = 3; in TEST() local
568 const int input_rows = 7; in TEST() local
635 const int input_rows = 7; in TEST() local
[all …]
Ddilation_ops.cc75 const int input_rows = input.dim_size(1); in ParseSizes() local
169 const int input_rows = input.dimension(1); in operator ()() local
278 const int input_rows = input.dimension(1); in operator ()() local
398 const int input_rows = input.dimension(1); in operator ()() local
Deigen_benchmark.h87 Eigen::Index input_rows = input_dims[1]; in SpatialConvolutionBackwardInput() local
220 Eigen::Index input_rows = input_dims[1]; in CuboidConvolutionBackwardInput() local
Dconv_ops.cc222 const Tensor& filter, int batch, int input_rows, in Run() argument
238 const Tensor& filter, int batch, int input_rows, in Run()
279 const Tensor& filter, int batch, int input_rows, in Run()
293 const Tensor& filter, int batch, int input_rows, in Run()
441 const int input_rows = static_cast<int>(input_rows_raw); in ComputeConv2DDimension() local
Deigen_spatial_convolutions_test.cc31 const int input_rows = 4; in TEST() local
76 const int input_rows = 4; in TEST() local
195 const int input_rows = 5; in TEST() local
247 const int input_rows = 5; in TEST() local
305 const int input_rows = 5; in TEST() local
361 const int input_rows = 5; in TEST() local
409 const int input_rows = 3; in TEST() local
457 const int input_rows = 5; in TEST() local
508 const int input_rows = 7; in TEST() local
559 const int input_rows = 7; in TEST() local
[all …]
Ddepthwise_conv_op.cc360 const int32 input_rows = static_cast<int32>(input_rows_raw); in Compute() local
Dconv_ops_using_gemm.cc497 const int input_rows = static_cast<int>(input_rows_raw); in Compute() local
Dquantized_conv_ops.cc524 const int64 input_rows = input.dim_size(1); in Compute() local
/external/libopus/src/
Dmapping_matrix.c88 int input_rows, in mapping_matrix_multiply_channel_in_float()
123 int input_rows, in mapping_matrix_multiply_channel_out_float()
159 int input_rows, in mapping_matrix_multiply_channel_in_short()
200 int input_rows, in mapping_matrix_multiply_channel_out_short()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc26 tensorflow::int64 input_rows, tensorflow::int64 input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
47 tensorflow::int64 input_batch, tensorflow::int64 input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc28 tensorflow::int64 input_batch, tensorflow::int64 input_rows, in __xla_cpu_runtime_EigenConvF32()
52 tensorflow::int64 input_rows, tensorflow::int64 input_cols, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc53 ScalarType* rhs, int64 input_batch, int64 input_rows, in MKLConvImpl()
156 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h34 Eigen::Index input_rows, Eigen::Index input_cols, in EigenConvImpl()
Dir_emitter.cc903 int64 input_rows = in HandleConvolution() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h138 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder() local
162 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder() local
330 int input_planes, input_rows, input_cols; variable
/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch.cpp186 int input_rows = 3; in test_patch_padding_valid() local
262 int input_rows = 5; in test_patch_padding_valid_same_value() local
335 int input_rows = 4; in test_patch_padding_same() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc79 const int32 input_rows = input.dim_size(1); in Compute() local
/external/libjpeg-turbo/
Djcprepct.c109 expand_bottom_edge(JSAMPARRAY image_data, JDIMENSION num_cols, int input_rows, in expand_bottom_edge()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1852 input_rows, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h661 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected() local
2879 const int input_rows = input_dims.sizes[0]; in ConvAsGemm() local
Doptimized_ops.h340 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected() local