Home
last modified time | relevance | path

Searched defs:kernel_size (Results 1 – 25 of 63) 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()
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.py85 kernel_size, argument
125 kernel_size, argument
290 kernel_size, argument
330 kernel_size, argument
503 kernel_size, argument
543 kernel_size, argument
718 kernel_size, argument
833 kernel_size, argument
882 kernel_size, argument
1005 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.py226 def block1(x, filters, kernel_size=3, stride=1, conv_shortcut=True, name=None): argument
290 def block2(x, filters, kernel_size=3, stride=1, conv_shortcut=False, name=None): argument
361 kernel_size=3, argument
Dinception_resnet_v2.py255 kernel_size, argument
Dimagenet_utils.py388 def correct_pad(inputs, 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/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py116 kernel_size, argument
486 kernel_size, argument
645 kernel_size, argument
792 kernel_size, argument
920 kernel_size, argument
1191 kernel_size, argument
1501 kernel_size, argument
1766 kernel_size, argument
1999 kernel_size, argument
2189 kernel_size, argument
[all …]
Dlocal.py121 kernel_size, argument
423 kernel_size, argument
Dconvolutional_recurrent.py493 kernel_size, argument
830 kernel_size, argument
897 def kernel_size(self): member in ConvLSTM2D
/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/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()
Dextract_image_patches_op.cc108 int64 kernel_size = 1; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc47 int64 kernel_size = option[2]; in GetConv2DTestCases() local
/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/test/
Dconv-hwc-microkernel-tester.h161 inline ConvHWCMicrokernelTester& kernel_size(uint32_t kernel_size) { in kernel_size() function
Dconv-hwc2chw-microkernel-tester.h161 inline ConvHWC2CHWMicrokernelTester& kernel_size(uint32_t kernel_size) { in kernel_size() function
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator_test.cc55 const int kernel_size = 3; in CreateMiniGraph() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_thin.cc54 const int2& kernel_size) { in GenerateConvolutionTransposedCode()
/external/XNNPACK/bench/
Df32-conv-hwc2chw.cc44 const size_t kernel_size = 3; in DConvHWC2CHW3X3S2P1Benchmark() local

123