Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_to_tensor_op_test.py508 rt_value = [[1, 2, 3], [4], [5, 6]],
517 rt_value = [[1, 2, 3], [4], [5, 6]],
528 rt_value = [[1, 2, 3], [4], [5, 6, 7, 8, 9], [8]],
539 rt_value = [[[9, 8, 7], [6]], [[5, 4]]],
549 rt_value = [[[9, 8], [7, 6]], [[5, 4]]],
559 rt_value = [[[9, 8], [7, 6]], [[5, 4]]],
569 rt_value = [[[9, 8], [7, 6]], [[5, 4]]],
579 rt_value = [[[[9], [8]], [[7], [6]]], [[[5], [4]]]],
593 rt_value = [[[[9], [8]], [[7], [6]]], [[[5], [4]]]],
606 rt_value = [[9, 8], [7, 6, 5], [4]],
[all …]
Dragged_tensor_test.py122 rt_value = ragged_tensor_value.RaggedTensorValue(values, splits)
123 self.assertEqual(rt_value.row_splits.dtype, np.int64)
124 self.assertEqual(rt_value.shape, (5, None))
125 self.assertLen(rt_value.nested_row_splits, 1)
126 self.assertAllEqual(splits, rt_value.row_splits)
127 self.assertAllEqual(values, rt_value.values)
128 self.assertAllEqual(splits, rt_value.nested_row_splits[0])
129 self.assertAllEqual(values, rt_value.flat_values)
132 rt_value = ragged_tensor_value.RaggedTensorValue(
135 self.assertEqual(rt_value.row_splits.dtype, np.int64)
[all …]
/external/tensorflow/tensorflow/python/framework/
Dtensor_util_test.py811 rt_value = self.evaluate(rt)
813 self.assertFalse(tensor_util.is_tf_type(rt_value))