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.cc178 for (int64 unknown_dim : unknown_dims) { in CompareSymbolicallyShapedTensorSizes() local
179 int64 co_occurrence = std::min(left_unknown_dims[unknown_dim], in CompareSymbolicallyShapedTensorSizes()
180 right_unknown_dims[unknown_dim]); in CompareSymbolicallyShapedTensorSizes()
181 left_unknown_dims[unknown_dim] -= co_occurrence; in CompareSymbolicallyShapedTensorSizes()
182 right_unknown_dims[unknown_dim] -= co_occurrence; in CompareSymbolicallyShapedTensorSizes()
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py407 unknown_dim = [False]
412 unknown_dim[0] = True
416 self.assertFalse(unknown_dim[0])
420 self.assertFalse(unknown_dim[0])
424 self.assertTrue(unknown_dim[0])
430 unknown_dim = [False]
437 unknown_dim[0] = True
442 self.assertFalse(unknown_dim[0])
445 self.assertFalse(unknown_dim[0])
448 self.assertTrue(unknown_dim[0])
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py2265 unknown_dim = array_ops.placeholder(dtype=dtypes.int32)
2272 array_ops.zeros([unknown_dim], dtype=dtypes.int64))
2277 [batch_size, unknown_dim], dtype=dtypes.int64))
2294 output_vals = sess.run(output, feed_dict={unknown_dim: 1})