Home
last modified time | relevance | path

Searched refs:unknown_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes.cc175 for (int64 unknown_dim : unknown_dims) { in CompareSymbolicallyShapedTensorSizes() local
176 int64 co_occurrence = std::min(left_unknown_dims[unknown_dim], in CompareSymbolicallyShapedTensorSizes()
177 right_unknown_dims[unknown_dim]); in CompareSymbolicallyShapedTensorSizes()
178 left_unknown_dims[unknown_dim] -= co_occurrence; in CompareSymbolicallyShapedTensorSizes()
179 right_unknown_dims[unknown_dim] -= co_occurrence; in CompareSymbolicallyShapedTensorSizes()
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py133 unknown_dim = [False]
138 unknown_dim[0] = True
142 self.assertFalse(unknown_dim[0])
146 self.assertFalse(unknown_dim[0])
150 self.assertTrue(unknown_dim[0])
154 unknown_dim = [False]
161 unknown_dim[0] = True
167 self.assertFalse(unknown_dim[0])
171 self.assertFalse(unknown_dim[0])
175 self.assertTrue(unknown_dim[0])
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py2255 unknown_dim = array_ops.placeholder(dtype=dtypes.int32)
2262 array_ops.zeros([unknown_dim], dtype=dtypes.int64))
2267 [batch_size, unknown_dim], dtype=dtypes.int64))
2284 output_vals = sess.run(output, feed_dict={unknown_dim: 1})