Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.cc225 const std::vector<int32>& ksize, in Init()
253 const std::vector<int32>& ksize, in Init()
268 const std::vector<int32>& ksize, in Init()
293 const std::vector<int32>& ksize, in Init()
Dquantized_pooling_ops_test.cc39 const int ksize = 2; in TEST_F() local
84 const int ksize = 2; in TEST_F() local
Dmkl_quantized_pooling_ops_test.cc65 const int ksize = 2; in TEST_F() local
134 const int ksize = 2; in TEST_F() local
Dmaxpooling_op.cc270 std::vector<int32> ksize = ksize_; in Compute() local
394 std::vector<int32> ksize = ksize_; in Compute() local
499 std::vector<int32> ksize = ksize_; in Compute() local
700 std::vector<int32> ksize = ksize_; in Compute() local
838 std::vector<int32> ksize = ksize_; in Compute() local
1241 std::vector<int32> ksize = ksize_; in Compute() local
Dpooling_ops_common.cc47 const std::vector<int32>& ksize, in PoolParameters()
Dpooling_ops_3d.cc54 const std::vector<int32>& ksize, in Pool3dParameters()
Dops_util_test.cc58 int ksize; // Kernel size. member
/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/framework/
Dops_util.cc40 Status GetBroadcastSize(const int index, const int in_size, const int ksize, in GetBroadcastSize()
/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/linux-kselftest/tools/testing/selftests/bpf/
Dtest_get_stack_rawtp.c60 int max_len, max_buildid_len, usize, ksize, total_size; in bpf_prog1() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py3368 def avg_pool_v2(input, ksize, strides, padding, data_format=None, name=None): # pylint: disable=re… argument
3430 def avg_pool(value, ksize, strides, padding, data_format="NHWC", argument
3474 def avg_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): # pylint: disable=r… argument
3513 def avg_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): # pylint: disable=re… argument
3558 def avg_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): # pylint: disable=… argument
3598 def max_pool_v2(input, ksize, strides, padding, data_format=None, name=None): argument
3659 ksize, argument
3703 def max_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): argument
3747 def max_pool2d(input, ksize, strides, padding, data_format="NHWC", name=None): argument
3785 def max_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): argument
[all …]
Dinit_ops.py873 def _orthogonal_kernel(self, ksize, cin, cout): argument
998 def _orthogonal_kernel(self, ksize, cin, cout): argument
1147 def _orthogonal_kernel(self, ksize, cin, cout): argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py108 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
185 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
219 ksize, argument
735 ksize, argument
766 def _CompareMaxPoolingFwd(self, input_shape, ksize, strides, padding): argument
779 def _CompareMaxPoolingBk(self, input_shape, output_shape, ksize, strides, argument
809 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.cc91 std::vector<int64> ksize; in GetKernelSize() local
164 std::vector<int64> ksize = ksize_or_error.ValueOrDie(); in Compile() local
217 std::vector<int64> ksize = ksize_or_error.ValueOrDie(); in Compile() 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/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/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/core/grappler/costs/
Dop_level_cost_estimator_test.cc299 const std::vector<int>& ksize, in GetPoolingOutputSize()
369 const std::vector<int>& ksize, in DescribePoolingOp()
Dop_level_cost_estimator.cc129 const auto ksize = op_info.attr().at("ksize").list().i(); in GetKernelSize() local
1466 std::vector<int64> ksize = GetKernelSize(op_info); in OpDimensionsFromInputs() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc988 auto& ksize = (*maxpool_op->mutable_attr())["ksize"]; in ConvertMaxPoolOperator() local
1016 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertAveragePoolOperator() local
1093 auto& ksize = (*avgpool_op->mutable_attr())["ksize"]; in ConvertL2PoolOperator() local

12