Home
last modified time | relevance | path

Searched defs:kernel_size (Results 1 – 25 of 85) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc35 XlaOp input, absl::Span<const int64> kernel_size, in MakeGeneralPadding()
71 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
84 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
97 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
111 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
127 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
143 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
161 auto kernel_size = ExpandWithBatchAndFeatureDimensions({3, 3}, data_format); in XLA_TEST_F() local
176 auto kernel_size = ExpandWithBatchAndFeatureDimensions({3, 3}, data_format); in XLA_TEST_F() local
190 auto kernel_size = ExpandWithBatchAndFeatureDimensions({2, 2}, data_format); in XLA_TEST_F() local
[all …]
Dpooling.cc74 absl::Span<const int64> kernel_size, in ComputeSums()
134 XlaOp MaxPool(XlaOp operand, absl::Span<const int64> kernel_size, in MaxPool()
148 XlaOp AvgPool(XlaOp operand, absl::Span<const int64> kernel_size, in AvgPool()
174 absl::Span<const int64> input_size, absl::Span<const int64> kernel_size, in MakeSpatialPadding()
194 absl::Span<const int64> kernel_size, in AvgPoolGrad()
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py86 kernel_size, argument
127 kernel_size, argument
293 kernel_size, argument
334 kernel_size, argument
508 kernel_size, argument
549 kernel_size, argument
725 kernel_size, argument
841 kernel_size, argument
891 kernel_size, argument
1015 kernel_size, argument
[all …]
/external/gemmlowp/meta/generators/
Dtransform_kernels_common.py38 def Check(self, in_type, out_type, kernel_size, leftovers): argument
82 def Check(self, in_type, out_type, kernel_size, leftovers): argument
151 def Check(self, in_type, out_type, kernel_size, leftovers): argument
220 def Check(self, in_type, out_type, kernel_size, leftovers): argument
313 def EmitTransform(self, in_type, out_type, kernel_size, leftovers): argument
399 def EmitTransform(self, in_type, out_type, kernel_size, leftovers): argument
457 def _ProcessRow(self, emitter, registers, kernel_size, leftovers): argument
/external/ComputeLibrary/src/runtime/CL/functions/
DCLWinogradConvolutionLayer.cpp83 bool check_support_fast_math(const Size2D &output_tile, const Size2D &kernel_size) in check_support_fast_math()
125 …const Size2D kernel_size = Size2D(weights->info()->tensor_shape()[idx_width], weights->info()->ten… in configure() local
177 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in validate() local
DCLFFTConvolutionLayer.cpp129 …const Size2D kernel_size = Size2D(weights->info()->tensor_shape()[idx_width], weights->info()->ten… in configure() local
280 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in validate() local
/external/ComputeLibrary/src/core/CL/kernels/
DCLDirectConvolutionLayerKernel.cpp105 …arget gpu_target, unsigned int conv_stride_x, unsigned int conv_stride_y, unsigned int kernel_size, in can_run_optimized_kernel_for_bifrost()
118 …arget gpu_target, unsigned int conv_stride_x, unsigned int conv_stride_y, unsigned int kernel_size, in can_run_optimized_kernel_for_bifrost_nhwc()
133unsigned int kernel_size, const PadStrideInfo &conv_info, const GPUTarget target, ITensorInfo *inp… in setup_num_elems()
344 const unsigned int kernel_size = weights->dimension(width_idx); in validate_and_configure_window() local
427 const unsigned int kernel_size = weights->info()->dimension(width_idx); in configure() local
DCLWinogradInputTransformKernel.cpp52 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments() local
84 const Size2D kernel_size = winograd_info.kernel_size; in validate_and_configure_window() local
122 const Size2D kernel_size = winograd_info.kernel_size; in configure() local
DCLWinogradFilterTransformKernel.cpp55 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments() local
118 const Size2D kernel_size = winograd_info.kernel_size; in configure() local
DCLWinogradOutputTransformKernel.cpp62 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments() local
151 const Size2D kernel_size = winograd_info.kernel_size; in configure() local
/external/ComputeLibrary/tests/validation/fixtures/
DDepthwiseConvolutionLayerFixture.h59 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
194 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
208 …void setup(size_t width, size_t height, size_t channel, size_t batch, Size2D kernel_size, size_t d… in setup()
317 …void setup(size_t width, size_t height, size_t channel, size_t batch, Size2D kernel_size, size_t d… in setup()
438 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
452 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
DDirectConvolutionLayerFixture.h55 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
202 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
215 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
DDirectConvolutionLayerTensorShiftFixture.h52 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
210 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
222 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
/external/vboot_reference/futility/
Dvb1_helper.c129 static int KernelSize(uint8_t *kernel_buf, uint64_t kernel_size, in KernelSize()
152 static int PickApartVmlinuz(uint8_t *kernel_buf, uint64_t kernel_size, in PickApartVmlinuz()
397 uint8_t *SignKernelBlob(uint8_t *kernel_blob, uint64_t kernel_size, in SignKernelBlob()
491 uint64_t kernel_size, in VerifyKernelBlob()
/external/tensorflow/tensorflow/python/keras/applications/
Dresnet.py212 def block1(x, filters, kernel_size=3, stride=1, conv_shortcut=True, name=None): argument
276 def block2(x, filters, kernel_size=3, stride=1, conv_shortcut=False, name=None): argument
347 kernel_size=3, argument
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFFTConvolutionLayer.cpp122 …const Size2D kernel_size = Size2D(weights->info()->tensor_shape()[idx_width], weights->info()->ten… in configure() local
273 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in validate() local
/external/ComputeLibrary/src/runtime/NEON/functions/assembly/
DNEDepthwiseConvolutionAssemblyDispatch.cpp48 std::unique_ptr<depthwise::IDepthwiseConvolution> get_qasymm8_convolver(int kernel_size, int stride… in get_qasymm8_convolver()
90 std::unique_ptr<depthwise::IDepthwiseConvolution> get_qsymm8_perchannel_convolver(int kernel_size, … in get_qsymm8_perchannel_convolver()
133 std::unique_ptr<depthwise::IDepthwiseConvolution> get_fp16_convolver(int kernel_size, int stride_x, in get_fp16_convolver()
174 std::unique_ptr<depthwise::IDepthwiseConvolution> get_fp32_convolver(int kernel_size, int stride_x, in get_fp32_convolver()
239 const unsigned int kernel_size = weights->info()->tensor_shape().y(); in create_convolver() local
/external/ComputeLibrary/tests/validation/reference/
DWinograd.cpp45 …transform(SimpleTensor<T> &src, const Size2D &output_tile_size, const Size2D &kernel_size, Winogra… in initialize_matrix_transform()
242 const Size2D kernel_size = winograd_info.kernel_size; in winograd_input_transform() local
368 const Size2D kernel_size = winograd_info.kernel_size; in winograd_filter_transform() local
445 const Size2D kernel_size = winograd_info.kernel_size; in winograd_output_transform() local
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50.py46 def __init__(self, kernel_size, filters, stage, block, data_format): argument
105 kernel_size, argument
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py111 kernel_size, argument
488 kernel_size, argument
648 kernel_size, argument
795 kernel_size, argument
923 kernel_size, argument
1194 kernel_size, argument
1504 kernel_size, argument
1769 kernel_size, argument
2002 kernel_size, argument
2193 kernel_size, argument
[all …]
Dlocal_test.py530 kernel_size, argument
566 def get_model_saveable(implementation, filters, kernel_size, strides, layers, argument
/external/mesa3d/src/intel/vulkan/
Danv_pipeline_cache.c38 const void *kernel_data, uint32_t kernel_size, in anv_shader_bin_create()
205 uint32_t kernel_size = blob_read_uint32(blob); in anv_shader_bin_create_from_blob() local
416 uint32_t kernel_size, in anv_pipeline_cache_add_shader_locked()
447 const void *kernel_data, uint32_t kernel_size, in anv_pipeline_cache_upload_kernel()
708 const void *kernel_data, uint32_t kernel_size, in anv_device_upload_kernel()
/external/ComputeLibrary/src/runtime/GLES_COMPUTE/functions/
DGCDirectConvolutionLayer.cpp45 int kernel_size = weights->info()->dimension(0); in configure() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc81 std::vector<int64> kernel_size; // k member
126 int64_t kernel_size, int64_t stride) { in CalculateUpperPadding()
178 absl::Span<const int64> kernel_size, in MakeGeneralResizeKernel()
196 absl::Span<const int64> kernel_size, in MakeGeneralResizeKernelInDim()
/external/igt-gpu-tools/lib/
Dmedia_fill.c226 size_t kernel_size) in __gen9_media_fillfunc()
295 size_t kernel_size) in __gen11_media_vme_func()

1234