Home
last modified time | relevance | path

Searched refs:f_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dconv1d_transpose_test.py44 f_shape = [3, 2, 3]
49 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
72 f_shape = [3, 2, 3]
77 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
83 for k in xrange(f_shape[1]):
101 f_shape = [3, 2, 3]
106 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
117 for k in xrange(f_shape[1]):
136 f_shape = [3, 2, 3]
141 f_val = np.random.random_sample(f_shape).astype(np.float64)
[all …]
Dconv2d_transpose_test.py48 f_shape = [3, 3, 2, 3]
53 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
65 for k in xrange(f_shape[2]):
86 f_shape = [3, 3, 2, 3]
91 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
97 for k in xrange(f_shape[2]):
119 f_shape = [3, 3, 2, 3]
124 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
135 for k in xrange(f_shape[2]):
161 f_shape = [3, 3, 2, 3]
[all …]
Dconv3d_transpose_test.py44 f_shape = [3, 3, 3, 2, 3]
49 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
68 for k in xrange(f_shape[3]):
94 f_shape = [3, 3, 3, 2, 3]
99 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
105 for k in xrange(f_shape[3]):
127 f_shape = [3, 3, 3, 2, 2]
132 f_value = np.random.random_sample(f_shape).astype(np.float64)
142 f_shape = [3, 3, 3, 2, 3]
147 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
[all …]
Datrous_conv2d_test.py74 f_shape = [kernel_height, kernel_width, 2, 2]
75 f = np.arange(np.prod(f_shape), dtype=np.float32).reshape(f_shape)
121 f_shape = [kernel, kernel, 2, 2]
122 f = 1e-2 * np.random.random_sample(f_shape).astype(np.float32)
147 f_shape = [3, 3, 2, 2]
153 f_val = np.random.random_sample(f_shape).astype(np.float32)
160 [x_shape, f_shape],
181 f_shape = [kernel_height, kernel_width, 2, 2]
182 f = np.arange(np.prod(f_shape), dtype=np.float32).reshape(f_shape)
222 f_shape = [kernel_height, kernel_width, 2, 2]
[all …]
Datrous_convolution_test.py217 f_shape = [kernel, kernel, 2, 2]
218 f1 = 1e-2 * np.random.random_sample(f_shape).astype(np.float32)
219 f2 = 1e-2 * np.random.random_sample(f_shape).astype(np.float32)
254 def _test_gradient(self, x_shape, f_shape, dilation_rate, padding): argument
256 f_val = np.random.random_sample(f_shape).astype(np.float32)
262 err = gradient_checker.compute_gradient_error([x, f], [x_shape, f_shape],
275 f_shape=[3, 3, 2, 2],
Dconv1d_test.py64 f_shape = [3, 2, 3]
69 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
80 for k in xrange(f_shape[1]):
/external/tensorflow/tensorflow/compiler/tests/
Dconv3d_test.py80 f_shape = [3, 3, 3, 2, 3]
85 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
104 for k in xrange(f_shape[3]):
130 f_shape = [3, 3, 3, 2, 3]
135 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
141 for k in xrange(f_shape[3]):
168 f_shape = [3, 3, 3, 2, 3]
173 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
184 for k in xrange(f_shape[3]):
214 f_shape = [3, 3, 3, 2, 2]
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_test.py114 f_shape = array_ops.shape(f)
116 ops.add_to_collection('train_op', f_shape)
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3384 auto f_shape = make_shape(f_dims, options.filter_minor_to_major_layout); in TEST_F() local
3390 b.AddInstruction(HloInstruction::CreateParameter(1, f_shape, "filter")); in TEST_F()