Home
last modified time | relevance | path

Searched defs:ksize (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py74 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
110 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
299 ksize, argument
450 ksize, argument
554 def AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding, argument
Dpooling_ops_3d_test.py34 def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding): argument
194 ksize, argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc226 const std::vector<int32>& ksize, in Init()
253 const std::vector<int32>& ksize, in Init()
278 const std::vector<int32>& ksize, in Init()
Dmkl_quantized_pooling_ops_test.cc65 const int ksize = 2; in TEST_F() local
134 const int ksize = 2; in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/
Dquantized_pooling_ops_test.cc39 const int ksize = 2; in TEST_F() local
84 const int ksize = 2; in TEST_F() local
Dmaxpooling_op.cc286 std::vector<int32> ksize = ksize_; in Compute() local
396 std::vector<int32> ksize = ksize_; in Compute() local
502 std::vector<int32> ksize = ksize_; in Compute() local
708 std::vector<int32> ksize = ksize_; in Compute() local
860 std::vector<int32> ksize = ksize_; in Compute() local
1291 std::vector<int32> ksize = ksize_; in Compute() local
Dpooling_ops_common.cc117 const std::vector<int32>& ksize, in PoolParameters()
Dpooling_ops_3d.cc49 const std::vector<int32>& ksize, in Pool3dParameters()
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad_test.cc241 const std::vector<int> ksize{1, 2, 2, 1}; in TEST_F() local
254 Tensor ksize = test::AsTensor<int>({1, 2, 2, 1}, {4}); in TEST_F() local
267 const std::vector<int> ksize{1, 3, 3, 3, 1}; in TEST_F() local
280 const std::vector<int> ksize{1, 2, 2, 1}; in TEST_F() local
291 const std::vector<int> ksize{1, 3, 3, 3, 1}; in TEST_F() local
Dnn_grad.cc246 std::vector<int32> ksize; in MaxPoolGradHelper() local
281 std::vector<int32> ksize; in MaxPool3DGradHelper() local
302 std::vector<int32> ksize; in AvgPoolGradHelper() local
323 std::vector<int32> ksize; in AvgPool3DGradHelper() local
/external/tensorflow/tensorflow/core/framework/
Dops_util.cc40 Status GetBroadcastSize(const int index, const int in_size, const int ksize, in GetBroadcastSize()
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dpool.py119 def make_l2_pool(input_tensor, ksize, strides, padding, data_format): argument
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py4245 def avg_pool_v2(input, ksize, strides, padding, data_format=None, name=None): # pylint: disable=re… argument
4308 def avg_pool(value, ksize, strides, padding, data_format="NHWC", argument
4353 def avg_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): # pylint: disable=r… argument
4393 def avg_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): # pylint: disable=re… argument
4439 def avg_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): # pylint: disable=… argument
4480 def max_pool_v2(input, ksize, strides, padding, data_format=None, name=None): argument
4557 ksize, argument
4618 def max_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): argument
4676 def max_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): argument
4727 def max_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): argument
[all …]
Dinit_ops.py847 def _orthogonal_kernel(self, ksize, cin, cout): argument
975 def _orthogonal_kernel(self, ksize, cin, cout): argument
1131 def _orthogonal_kernel(self, ksize, cin, cout): argument
/external/linux-kselftest/tools/testing/selftests/bpf/progs/
Dtest_get_stack_rawtp.c60 int max_len, max_buildid_len, usize, ksize, total_size; in bpf_prog1() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py118 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
201 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
247 ksize, argument
888 ksize, argument
919 def _CompareMaxPoolingFwd(self, input_shape, ksize, strides, padding): argument
935 def _CompareMaxPoolingBk(self, input_shape, output_shape, ksize, strides, argument
968 def _CompareMaxPoolingGradBk(self, input_shape, output_shape, ksize, strides, argument
/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch.cpp189 int ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid() local
265 int ksize = 3; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid_same_value() local
338 int ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_same() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc93 std::vector<int64> ksize; in GetKernelSize() local
173 std::vector<int64> ksize = ksize_or_error.ValueOrDie(); in Compile() local
225 std::vector<int64> ksize = ksize_or_error.ValueOrDie(); in Compile() local
/external/llvm-project/clang/test/Sema/
Dvla.c105 const int ksize = 4; in test_fold_to_constant_array() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer_test.cc124 const gtl::ArraySlice<int>& ksize, in BuildMaxPoolOps()
183 const std::vector<int> ksize{1, 1, 1, 1}; in CreatePoolGraphDef() local
/external/eigen/bench/
Dbenchmark-blocking-sizes.cpp572 for (size_t ksize = minsize; ksize <= maxsize; ksize *= 2) { in run() local
603 for (size_t ksize = minsize; ksize <= maxsize; ksize *= 2) { in run() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc31 absl::Span<const int64> ksize, absl::Span<const int64> stride, in AvgPoolDivideByCountWithGeneralPadding()
/external/python/cpython2/PC/
D_subprocess.c359 size_t ksize, vsize, totalsize; in getenvironment() local
/external/tensorflow/tensorflow/python/ops/risc/
Drisc_ops.py190 def risc_pool(x, ksize, strides, pooling_type='MAX', name='RISC_POOL'): argument
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc996 auto& ksize = (*maxpool_op->mutable_attr())["ksize"]; in ConvertMaxPoolOperator() local
1024 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertAveragePoolOperator() local
1101 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertL2PoolOperator() local

12