| /external/tensorflow/tensorflow/python/kernel_tests/ |
| conv3d_transpose_test.py | 73 w_in = w > 0 and w < y_shape[3] - 1
74 if d_in + h_in + w_in == 3:
76 elif d_in + h_in + w_in == 2:
78 elif d_in or h_in or w_in:
111 w_in = w % strides[3] == 0 and 0 < w < y_shape[3] - 1
112 if d_in + h_in + w_in == 3:
114 elif d_in + h_in + w_in == 2:
116 elif d_in or h_in or w_in:
154 w_in = w % strides[3] == 0 and pad < w < y_shape[3] - 1 - pad
155 if d_in + h_in + w_in == 3
[all...] |
| conv2d_transpose_test.py | 69 w_in = w > 0 and w < y_shape[2] - 1
70 if h_in and w_in:
72 elif h_in or w_in:
102 w_in = w % strides[2] == 0 and w > 0 and w < y_shape[2] - 1
103 if h_in and w_in:
105 elif h_in or w_in:
141 w_in = w % strides[2] == 0 and w > pad and w < y_shape[
143 if h_in and w_in:
145 elif h_in or w_in:
204 w_in = w > 0 and w < y_shape[3] -
[all...] |
| conv1d_test.py | 86 w_in = w % stride == 0 and w > pad and w < y_shape[1] - 1 - pad
87 if w_in:
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| conv3d_test.py | 110 w_in = w > 0 and w < y_shape[3] - 1
111 if d_in + h_in + w_in == 3:
113 elif d_in + h_in + w_in == 2:
115 elif d_in or h_in or w_in:
148 w_in = w % strides[3] == 0 and 0 < w < y_shape[3] - 1
149 if d_in + h_in + w_in == 3:
151 elif d_in + h_in + w_in == 2:
153 elif d_in or h_in or w_in:
191 w_in = w % strides[3] == 0 and pad < w < y_shape[3] - 1 - pad
192 if d_in + h_in + w_in == 3
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| dilation_ops_gpu.cu.cc | 61 const int w_in = w_beg + w * rate_cols; local
62 if (w_in >= 0 && w_in < input_cols) {
64 input_ptr[d + depth * (w_in +
105 const int w_in = w_beg + w * rate_cols; local
106 if (w_in >= 0 && w_in < input_cols) {
108 input_ptr[d + depth * (w_in +
114 w_in_max = w_in;
154 const int w_in = w_beg + w * rate_cols local
[all...] |
| dilation_ops.cc | 192 const int w_in = w_beg + w * rate_cols; local
193 if (w_in >= 0 && w_in < input_cols) {
194 const T val = input(b, h_in, w_in, d) + filter(h, w, d);
309 const int w_in = w_beg + w * rate_cols; local
310 if (w_in >= 0 && w_in < input_cols) {
311 const T val = input(b, h_in, w_in, d) + filter(h, w, d);
315 w_in_max = w_in;
429 const int w_in = w_beg + w * rate_cols local
[all...] |
| word2vec_kernels.cc | 263 Tensor w_in = ctx->mutable_input(0, false); variable
264 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(w_in.shape()),
267 OP_REQUIRES(ctx, w_in.shape() == w_out.shape(),
268 errors::InvalidArgument("w_in.shape == w_out.shape"));
279 auto Tw_in = w_in.matrix<float>();
284 const int64 vocab_size = w_in.dim_size(0);
285 const int64 dims = w_in.dim_size(1);
|
| /prebuilts/tools/common/m2/repository/com/squareup/haha/haha/1.3/ |
| haha-1.3.jar | |
