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.py49 f_shape = [3, 3, 2, 3]
52 f = constant_op.constant(1, shape=f_shape, name="filter", dtype=dtype)
65 for k in xrange(f_shape[2]):
90 f_shape = [3, 3, 2, 3]
93 f = constant_op.constant(1, shape=f_shape, name="filter", dtype=dtype)
99 for k in xrange(f_shape[2]):
126 f_shape = [3, 3, 2, 3]
129 f = constant_op.constant(1, shape=f_shape, name="filter", dtype=dtype)
140 for k in xrange(f_shape[2]):
169 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)
120 f_shape = [kernel, kernel, 2, 2]
121 f = 1e-2 * np.random.random_sample(f_shape).astype(np.float32)
145 f_shape = [3, 3, 2, 2]
151 f_val = np.random.random_sample(f_shape).astype(np.float32)
158 [x_shape, f_shape],
179 f_shape = [kernel_height, kernel_width, 2, 2]
180 f = np.arange(np.prod(f_shape), dtype=np.float32).reshape(f_shape)
219 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],
276 f_shape=[3, 3, 2, 2],
Dconv1d_test.py93 f_shape = [3, 2, 3]
98 1.0, shape=f_shape, name="filter", dtype=dtypes.float32)
109 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.py118 f_shape = array_ops.shape(f)
120 ops.add_to_collection('train_op', f_shape)
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc4145 auto f_shape = make_shape(f_dims, options.filter_minor_to_major_layout); in TEST_F() local
4150 b.AddInstruction(HloInstruction::CreateParameter(1, f_shape, "filter")); in TEST_F()
4152 in_shape, f_shape, /*feature_group_count=*/1, in TEST_F()