Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc35 XlaOp input, absl::Span<const int64_t> 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_t> kernel_size, in ComputeSums()
134 XlaOp MaxPool(XlaOp operand, absl::Span<const int64_t> kernel_size, in MaxPool()
148 XlaOp AvgPool(XlaOp operand, absl::Span<const int64_t> kernel_size, in AvgPool()
174 absl::Span<const int64_t> input_size, absl::Span<const int64_t> kernel_size, in MakeSpatialPadding()
194 absl::Span<const int64_t> kernel_size, in AvgPoolGrad()
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py82 kernel_size, argument
123 kernel_size, argument
289 kernel_size, argument
330 kernel_size, argument
504 kernel_size, argument
545 kernel_size, argument
721 kernel_size, argument
837 kernel_size, argument
887 kernel_size, argument
1011 kernel_size, argument
[all …]
/external/ComputeLibrary/tests/validation/fixtures/
DDepthwiseConvolutionLayerFixture.h293 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
307 …void setup(size_t width, size_t height, size_t channel, size_t batch, Size2D kernel_size, size_t d… in setup()
446 …void setup(size_t width, size_t height, size_t channel, size_t batch, Size2D kernel_size, size_t d… in setup()
644 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
658 …void setup(TensorShape in_shape, Size2D kernel_size, PadStrideInfo pad_stride_info, Size2D dilatio… in setup()
DDirectConvolutionLayerFixture.h234 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
247 … input_shape, int stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned… in setup()
/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/gpu/cl/kernels/
DClWinogradInputTransformKernel.cpp57 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments() local
89 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.cpp57 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments() local
124 const Size2D kernel_size = winograd_info.kernel_size; in configure() local
DClWinogradOutputTransformKernel.cpp66 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/src/gpu/cl/operators/
DClWinogradConv2d.cpp89 bool check_support_fast_math(const Size2D &output_tile, const Size2D &kernel_size) in check_support_fast_math()
115 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in validate_arguments() local
186 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in configure() local
/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/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
DDepthwiseConv2dFixture.h64 …void setup(TensorShape input_shape, Size2D kernel_size, const PadStrideInfo &pad_stride, const Siz… in setup()
217 …void setup(TensorShape input_shape, Size2D kernel_size, const PadStrideInfo &info, const Size2D &d… in setup()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFFTConvolutionLayer.cpp123 …const Size2D kernel_size = Size2D(weights->info()->tensor_shape()[idx_width], weights->info()->ten… in configure() local
276 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in validate() local
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50.py42 def __init__(self, kernel_size, filters, stage, block, data_format): argument
101 kernel_size, argument
/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/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 …]
/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/CL/functions/
DCLFFTConvolutionLayer.cpp132 …const Size2D kernel_size = Size2D(weights->info()->tensor_shape()[idx_width], weights->info()->ten… in configure() local
284 …const Size2D kernel_size = Size2D(weights->tensor_shape()[idx_width], weights->tensor_shape()[idx_… in validate() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dquantized_transpose_conv_tester.h178 int32_t ComputeInputSize(int32_t output_size, int32_t kernel_size, in ComputeInputSize()
193 int32_t ComputePadding(int32_t output_size, int32_t kernel_size, in ComputePadding()
Dtranspose_conv_tester.h186 int32_t ComputeInputSize(int32_t output_size, int32_t kernel_size, in ComputeInputSize()
201 int32_t ComputePadding(int32_t output_size, int32_t kernel_size, in ComputePadding()
/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()
/external/crosvm/x86_64/src/
Dbzimage.rs79 let kernel_size = (params.hdr.syssize as usize) in load_bzimage() localVariable
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc86 std::vector<int64_t> kernel_size; // k member
131 int64_t kernel_size, int64_t stride) { in CalculateUpperPadding()
183 absl::Span<const int64_t> kernel_size, in MakeGeneralResizeKernel()
201 absl::Span<const int64_t> kernel_size, in MakeGeneralResizeKernelInDim()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc48 int64_t kernel_size = option[2]; in GetConv2DTestCases() local
/external/gemmlowp/meta/
Dtest_transform_benchmark.cc46 #define kernel_size (16) macro

1234