Searched refs:sparse_read (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | autocast_variable.py | 109 def sparse_read(self, indices, name=None): member in AutoCastVariable
111 val = self._variable.sparse_read(indices, name=name)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | variable_ops_test.py | 78 x = v.sparse_read(2)
89 x = v.sparse_read([2, 1])
101 x = v.sparse_read([[2, 1], [0, 2]])
115 x = v.sparse_read([[2, 1], [3, 0]])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | resource_variable_ops_test.py | 697 value = self.evaluate(v.sparse_read([0, 3, 1, 2]))
850 self.assertEqual("(3, 20, 35)", str(v.sparse_read([0, 1, 2]).shape))
854 str(v.sparse_read(array_ops.placeholder(dtypes.int32)).shape))
1085 gather = v.sparse_read([0])
|
| D | control_flow_ops_py_test.py | 1049 lambda: x * math_ops.reduce_sum(var.sparse_read([1, 2])),
1072 y1 = var.sparse_read([1, 2])
2704 return x + math_ops.reduce_sum(var.sparse_read([1, 3]))
2725 return x + math_ops.reduce_sum(var.sparse_read([1, 3]))
2754 y + math_ops.reduce_sum(var.sparse_read([1, 2]))),
2776 return x + var.sparse_read([1])[0]
2805 lambda i, x: (i + 1, x * math_ops.reduce_sum(var.sparse_read([1, 3]))),
2828 y1 = var1.sparse_read([1, 3])
2830 y3 = x3 * math_ops.reduce_sum(var2.sparse_read([0]))
|
| D | list_ops_test.py | 926 v.sparse_read(0), dtypes.float32)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 711 return params.sparse_read(ids, name=name)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 885 def sparse_read(self, indices, name=None): member in ResourceVariable
|
| D | array_ops.py | 3373 return params.sparse_read(indices, name=name)
|