Home
last modified time | relevance | path

Searched refs:w_in (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dconv2d_transpose_test.py70 w_in = w > 0 and w < y_shape[2] - 1
71 if h_in and w_in:
73 elif h_in or w_in:
103 w_in = w % strides[2] == 0 and w > 0 and w < y_shape[2] - 1
104 if h_in and w_in:
106 elif h_in or w_in:
142 w_in = w % strides[2] == 0 and w > pad and w < y_shape[
144 if h_in and w_in:
146 elif h_in or w_in:
206 w_in = w > 0 and w < y_shape[3] - 1
[all …]
Dconv3d_transpose_test.py74 w_in = w > 0 and w < y_shape[3] - 1
75 if d_in + h_in + w_in == 3:
77 elif d_in + h_in + w_in == 2:
79 elif d_in or h_in or w_in:
112 w_in = w % strides[3] == 0 and 0 < w < y_shape[3] - 1
113 if d_in + h_in + w_in == 3:
115 elif d_in + h_in + w_in == 2:
117 elif d_in or h_in or w_in:
185 w_in = w % strides[3] == 0 and pad < w < y_shape[3] - 1 - pad
186 if d_in + h_in + w_in == 3:
[all …]
Dconv1d_transpose_test.py58 w_in = w > 0 and w < y_shape[1] - 1
59 if w_in:
87 w_in = w % strides[1] == 0 and w > 0 and w < y_shape[1] - 1
88 if w_in:
121 w_in = w % strides[1] == 0 and w > pad and w < y_shape[1] - 1 - pad
122 if w_in:
179 w_in = w > 0 and w < y_shape[2] - 1
180 if w_in:
211 w_in = w % strides[2] == 0 and w > 0 and w < y_shape[2] - 1
212 if w_in:
[all …]
Dconv1d_test.py84 w_in = w % stride == 0 and w > pad and w < y_shape[1] - 1 - pad
85 if w_in:
/external/tensorflow/tensorflow/compiler/tests/
Dconv3d_test.py110 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/
Ddilation_ops_gpu.cu.cc61 const int w_in = w_beg + w * rate_cols; in DilationKernel() local
62 if (w_in >= 0 && w_in < input_cols) { in DilationKernel()
64 input_ptr[d + depth * (w_in + in DilationKernel()
105 const int w_in = w_beg + w * rate_cols; in DilationBackpropInputKernel() local
106 if (w_in >= 0 && w_in < input_cols) { in DilationBackpropInputKernel()
108 input_ptr[d + depth * (w_in + in DilationBackpropInputKernel()
114 w_in_max = w_in; in DilationBackpropInputKernel()
154 const int w_in = w_beg + w * rate_cols; in DilationBackpropFilterKernel() local
155 if (w_in >= 0 && w_in < input_cols) { in DilationBackpropFilterKernel()
157 input_ptr[d + depth * (w_in + in DilationBackpropFilterKernel()
Ddilation_ops.cc192 const int w_in = w_beg + w * rate_cols; in operator ()() local
193 if (w_in >= 0 && w_in < input_cols) { in operator ()()
194 const T val = input(b, h_in, w_in, d) + filter(h, w, d); in operator ()()
309 const int w_in = w_beg + w * rate_cols; in operator ()() local
310 if (w_in >= 0 && w_in < input_cols) { in operator ()()
311 const T val = input(b, h_in, w_in, d) + filter(h, w, d); in operator ()()
315 w_in_max = w_in; in operator ()()
429 const int w_in = w_beg + w * rate_cols; in operator ()() local
430 if (w_in >= 0 && w_in < input_cols) { in operator ()()
431 const T val = input(b, h_in, w_in, d) + filter(h, w, d); in operator ()()
Dword2vec_kernels.cc263 Tensor w_in = ctx->mutable_input(0, false); in Compute() local
264 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(w_in.shape()), in Compute()
267 OP_REQUIRES(ctx, w_in.shape() == w_out.shape(), in Compute()
279 auto Tw_in = w_in.matrix<float>(); in Compute()
284 const int64 vocab_size = w_in.dim_size(0); in Compute()
285 const int64 dims = w_in.dim_size(1); in Compute()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_NegTrain.pbtxt4 name: "w_in"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt18792 name: "w_in"
18825 name: "w_in"
18859 name: "w_in"
Dops_history.v1.pbtxt41816 name: "w_in"
Dops_history.v2.pbtxt41319 name: "w_in"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt20726 name: "w_in"