Home
last modified time | relevance | path

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

123

/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()
133 XlaOp MaxPool(XlaOp operand, absl::Span<const int64> kernel_size, in MaxPool()
147 XlaOp AvgPool(XlaOp operand, absl::Span<const int64> kernel_size, in AvgPool()
173 absl::Span<const int64> input_size, absl::Span<const int64> kernel_size, in MakeSpatialPadding()
193 absl::Span<const int64> kernel_size, in AvgPoolGrad()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dpooling.cc37 std::string GetMaxPoolingCode(const HW& kernel_size) { in GetMaxPoolingCode()
82 std::string GetMaxPoolingIndicesCode(const HW& kernel_size) { in GetMaxPoolingIndicesCode()
145 std::string GetAveragePoolingCode(const HW& kernel_size) { in GetAveragePoolingCode()
Dmax_unpooling.cc37 std::string GetMaxUnpoolingCode(const HW& kernel_size) { in GetMaxUnpoolingCode()
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py81 kernel_size, argument
124 kernel_size, argument
280 kernel_size, argument
323 kernel_size, argument
487 kernel_size, argument
530 kernel_size, argument
696 kernel_size, argument
808 kernel_size, argument
860 kernel_size, argument
980 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/tensorflow/tensorflow/python/keras/applications/
Dresnet.py207 def block1(x, filters, kernel_size=3, stride=1, conv_shortcut=True, name=None): argument
271 def block2(x, filters, kernel_size=3, stride=1, conv_shortcut=False, name=None): argument
342 kernel_size=3, argument
Dimagenet_utils.py337 def correct_pad(inputs, kernel_size): argument
Dinception_resnet_v2.py235 kernel_size, argument
/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/eager/benchmarks/resnet50/
Dresnet50.py46 def __init__(self, kernel_size, filters, stage, block, data_format): argument
105 kernel_size, argument
/external/XNNPACK/src/
Dconvolution-nhwc.c48 size_t kernel_size, in find_dwigemm_ukernel()
230 const size_t kernel_size = kernel_height * kernel_width; in xnn_create_convolution2d_nhwc_q8() local
539 const size_t kernel_size = kernel_height * kernel_width; in xnn_create_convolution2d_nhwc_f32() local
860 const size_t kernel_size = kernel_height * kernel_width; in setup_convolution2d_nhwc() local
966 const size_t kernel_size = kernel_height * kernel_width; in setup_convolution2d_nhwc() local
Ddeconvolution-nhwc.c197 const uint32_t kernel_size = kernel_height * kernel_width; in xnn_create_deconvolution2d_nhwc_q8() local
446 const uint32_t kernel_size = kernel_height * kernel_width; in xnn_create_deconvolution2d_nhwc_f32() local
566 const size_t kernel_size = kernel_height * kernel_width; in setup_conv_path() local
671 const size_t kernel_size = kernel_height * kernel_width; in setup_subconv2d_path() local
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py104 kernel_size, argument
412 kernel_size, argument
565 kernel_size, argument
709 kernel_size, argument
856 kernel_size, argument
1150 kernel_size, argument
1407 kernel_size, argument
1635 kernel_size, argument
1820 kernel_size, argument
1968 kernel_size, argument
Dlocal.py136 kernel_size, argument
450 kernel_size, 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/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/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc47 int64 kernel_size = option[2]; in GetConv2DTestCases() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc81 std::vector<int64> kernel_size; // k member
124 int64 CalculateUpperPadding(int64 in_size, int64 out_size, int64 kernel_size, in CalculateUpperPadding()
177 absl::Span<const int64> kernel_size, in MakeGeneralResizeKernel()
195 absl::Span<const int64> kernel_size, in MakeGeneralResizeKernelInDim()
/external/crosvm/x86_64/src/
Dbzimage.rs88 let kernel_size = (params.hdr.syssize as usize) in load_bzimage() localVariable
/external/gemmlowp/meta/
Dtest_transform_benchmark.cc46 #define kernel_size (16) macro
Dmulti_thread_transform.h30 const Params& params, int kernel_size, in PrepareTransform1DTasks()
/external/XNNPACK/bench/
Df32-conv-hwc.cc43 const size_t kernel_size = 3; in DConv3X3S2P1Benchmark() local
Df32-conv-hwc2spchw.cc43 const size_t kernel_size = 3; in DConvHWC2SpCHW3X3S2P1Benchmark() local
/external/XNNPACK/test/
Dconv-hwc-microkernel-tester.h161 inline ConvHWCMicrokernelTester& kernel_size(uint32_t kernel_size) { in kernel_size() function

123