| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 74 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
98 ksize = NHWCToNCHW(ksize)
101 ksize=ksize,
110 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
124 self._VerifyOneTest(pool_func, input_sizes, ksize, strides, padding,
131 ksize=[1, 2, 2, 1],
140 ksize=[1, 2, 2, 1],
157 ksize=[1, 1, 2, 1],
166 ksize=[1, 2, 2, 1],
173 ksize=[1, 2, 2, 1],
[all …]
|
| D | pooling_ops_3d_test.py | 34 def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding): argument
39 ksize=ksize,
69 ksize=[1] + window + [1],
190 def _VerifyGradient(self, pool_func, pool_grad_func, input_sizes, ksize, argument
202 ksize = [1] + ksize + [1]
213 ksize=ksize,
231 ksize=ksize,
246 ksize=ksize,
268 ksize=[1, 1, 1],
277 ksize=[2, 2, 2],
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 95 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
126 input_sizes, total_size, pool_func, ksize, strides)
138 ksize = test_util.NHWCToNCHW(ksize)
145 ksize=ksize_placeholder,
152 ksize=ksize,
163 ksize_placeholder: ksize,
171 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
193 self._VerifyOneType(pool_func, input_sizes, ksize, strides, padding,
197 self._VerifyOneType(pool_func, input_sizes, ksize, strides, padding,
203 ksize, argument
[all …]
|
| D | pooling_ops_3d_test.py | 78 ksize=window,
248 ksize = [1, window[0], window[1], window[2], 1]
253 ksize = test_util.NHWCToNCHW(ksize)
259 ksize=ksize,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.cc | 29 const std::vector<int32>& ksize, in Init() argument
42 Init(context, ksize, stride, padding, data_format); in Init()
48 const std::vector<int32>& ksize, in Init() argument
58 Init(context, ksize, stride, padding, data_format); in Init()
63 const std::vector<int32>& ksize, in Init() argument
73 Init(context, ksize, stride, padding, data_format); in Init()
78 const std::vector<int32>& ksize, in Init() argument
85 window_rows = GetTensorDim(ksize, data_format, 'H'); in Init()
86 window_cols = GetTensorDim(ksize, data_format, 'W'); in Init()
87 depth_window = GetTensorDim(ksize, data_format, 'C'); in Init()
|
| D | ops_util.cc | 37 Status GetBroadcastSize(const int index, const int in_size, const int ksize, in GetBroadcastSize() argument
46 *bsize = ksize; in GetBroadcastSize()
50 *bsize = ksize + *bindex - pad_size; in GetBroadcastSize()
56 if (*bindex + ksize > in_size) { in GetBroadcastSize()
57 *bsize = std::min((in_size - *bindex), ksize); in GetBroadcastSize()
|
| D | maxpooling_op.cc | 256 std::vector<int32> ksize = ksize_; in Compute() local
261 ksize.resize(tensor_ksize.shape().num_elements()); in Compute()
262 std::copy_n(&value_ksize(0), ksize.size(), ksize.begin()); in Compute()
270 OP_REQUIRES(context, ksize.size() == 4, in Compute()
276 OP_REQUIRES(context, ksize[0] == 1 && stride[0] == 1, in Compute()
280 context, ksize[3] == 1 && stride[3] == 1, in Compute()
284 PoolParameters params{context, ksize, stride, in Compute()
380 std::vector<int32> ksize = ksize_; in Compute() local
385 ksize.resize(tensor_ksize.shape().num_elements()); in Compute()
386 std::copy_n(&value_ksize(0), ksize.size(), ksize.begin()); in Compute()
[all …]
|
| D | quantized_pooling_ops_test.cc | 39 const int ksize = 2; in TEST_F() local
46 .Attr("ksize", {1, ksize, ksize, 1}) in TEST_F()
84 const int ksize = 2; in TEST_F() local
91 .Attr("ksize", {1, ksize, ksize, 1}) in TEST_F()
|
| D | pooling_ops_common.h | 43 PoolParameters(OpKernelContext* context, const std::vector<int32>& ksize,
325 std::vector<int32> ksize = ksize_;
331 ksize.resize(tensor_ksize.shape().num_elements());
332 std::copy_n(&value_ksize(0), ksize.size(), ksize.begin());
340 OP_REQUIRES(context, ksize.size() == 4,
346 OP_REQUIRES(context, ksize[0] == 1 && stride[0] == 1,
350 PoolParameters params{context, ksize, stride,
|
| D | mkl_pooling_ops_common.h | 85 void Init(OpKernelContext* context, const std::vector<int32>& ksize,
89 void Init(OpKernelContext* context, const std::vector<int32>& ksize,
93 void Init(OpKernelContext* context, const std::vector<int32>& ksize,
100 void Init(OpKernelContext* context, const std::vector<int32>& ksize,
|
| D | pooling_ops_common.cc | 32 const std::vector<int32>& ksize, in PoolParameters() argument
51 window_rows = GetTensorDim(ksize, data_format, 'H'); in PoolParameters()
52 window_cols = GetTensorDim(ksize, data_format, 'W'); in PoolParameters()
53 depth_window = GetTensorDim(ksize, data_format, 'C'); in PoolParameters()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_image_patch.cpp | 189 int ksize = 2; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid() local
197 …Tensor<float, 5> result = tensor.extract_image_patches(ksize, ksize, stride, stride, 1, 1, PADDING… in test_patch_padding_valid()
200 VERIFY_IS_EQUAL(result.dimension(1), ksize); // kernel rows in test_patch_padding_valid()
201 VERIFY_IS_EQUAL(result.dimension(2), ksize); // kernel cols in test_patch_padding_valid()
212 …Tensor<float, 5, RowMajor> result_row_major = tensor_row_major.extract_image_patches(ksize, ksize,… in test_patch_padding_valid()
223 for (int i = 0; (i+stride+ksize-1) < input_rows; i += stride) { // input rows in test_patch_padding_valid()
224 for (int j = 0; (j+stride+ksize-1) < input_cols; j += stride) { // input cols in test_patch_padding_valid()
226 for (int r = 0; r < ksize; ++r) { // patch rows in test_patch_padding_valid()
227 for (int c = 0; c < ksize; ++c) { // patch cols in test_patch_padding_valid()
265 int ksize = 3; // Corresponds to the Rows and Cols for tensor.extract_image_patches<>. in test_patch_padding_valid_same_value() local
[all …]
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| D | parse_layer_parameters.py | 189 ksize = node.attr["ksize"]
190 kernel_size_y = ksize.list.i[1]
191 kernel_size_x = ksize.list.i[2]
192 if ksize.list.i[0] != 1:
194 if ksize.list.i[3] != 1:
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MaxPool3D.pbtxt | 16 name: "ksize"
19 the input tensor. Must have `ksize[0] = ksize[4] = 1`.
|
| D | api_def_AvgPool3D.pbtxt | 16 name: "ksize"
19 the input tensor. Must have `ksize[0] = ksize[4] = 1`.
|
| D | api_def_AvgPool3DGrad.pbtxt | 22 name: "ksize"
25 the input tensor. Must have `ksize[0] = ksize[4] = 1`.
|
| D | api_def_MaxPool3DGrad.pbtxt | 22 name: "ksize"
25 the input tensor. Must have `ksize[0] = ksize[4] = 1`.
|
| D | api_def_MaxPool3DGradGrad.pbtxt | 28 name: "ksize"
31 the input tensor. Must have `ksize[0] = ksize[4] = 1`.
|
| D | api_def_AvgPool.pbtxt | 16 name: "ksize"
45 Each entry in `output` is the mean of the corresponding size `ksize`
|
| /external/eigen/bench/ |
| D | benchmark-blocking-sizes.cpp | 572 for (size_t ksize = minsize; ksize <= maxsize; ksize *= 2) { in run() local
575 for (size_t kblock = minsize; kblock <= ksize; kblock *= 2) { in run()
578 benchmarks.emplace_back(ksize, msize, nsize, kblock, mblock, nblock); in run()
603 for (size_t ksize = minsize; ksize <= maxsize; ksize *= 2) { in run() local
606 benchmarks.emplace_back(ksize, msize, nsize); in run()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 82 std::vector<int64> ksize = ksize_; in Compile() local
94 ksize.clear(); in Compile()
95 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &ksize)); in Compile()
116 input, InitValue(ctx->builder(), type), *Reduction(ctx, type), ksize, in Compile()
180 const std::vector<int64>& ksize, const std::vector<int64>& stride, in AvgPoolDivideByCount() argument
186 int64 window_size = std::accumulate(ksize.begin(), ksize.end(), 1, in AvgPoolDivideByCount()
207 window_ksize[i] = ksize[dim]; in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad_test.cc | 185 const std::vector<int> ksize{1, 2, 2, 1}; in TEST_F() local
187 auto y = MaxPool(scope_, x, ksize, strides, "VALID"); in TEST_F()
198 Tensor ksize = test::AsTensor<int>({1, 2, 2, 1}, {4}); in TEST_F() local
200 auto y = MaxPoolV2(scope_, x, ksize, strides, "VALID"); in TEST_F()
|
| /external/python/cpython2/PC/ |
| D | _subprocess.c | 355 int ksize, vsize, totalsize; in getenvironment() local
364 ksize = PyString_GET_SIZE(key); in getenvironment()
366 totalsize = (p - PyString_AS_STRING(out)) + ksize + 1 + in getenvironment()
374 memcpy(p, PyString_AS_STRING(key), ksize); in getenvironment()
375 p += ksize; in getenvironment()
|
| /external/icu/icu4c/source/common/ |
| D | locid.cpp | 301 int32_t ksize = 0; in Locale() local
367 ksize = (int32_t)uprv_strlen(newKeywords); in Locale()
368 if ( ksize < 0 || ksize > ULOC_STRING_LIMIT ) { in Locale()
372 size += ksize + 1; in Locale()
396 if ( ksize != 0) in Locale()
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.nn.pbtxt | 25 …argspec: "args=[\'value\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'name\'], vararg…
29 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'name\'], vararg…
205 …argspec: "args=[\'value\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'name\'], vararg…
209 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'data_format\', \'name\'], vararg…
213 …argspec: "args=[\'input\', \'ksize\', \'strides\', \'padding\', \'Targmax\', \'name\'], varargs=No…
233 …argspec: "args=[\'input\', \'min_input\', \'max_input\', \'ksize\', \'strides\', \'padding\', \'na…
241 …argspec: "args=[\'input\', \'min_input\', \'max_input\', \'ksize\', \'strides\', \'padding\', \'na…
|