HomeSort by relevance Sort by last modified time
    Searched refs:_AsType (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tests/
scatter_nd_op_test.py 31 def _AsType(v, vtype):
102 indices = _AsType(indices[:num_updates], itype)
107 updates = _AsType(np.random.randn(*(updates_shape)), vtype)
  /external/tensorflow/tensorflow/python/kernel_tests/
scatter_ops_test.py 30 def _AsType(v, vtype):
92 updates = _AsType(
100 old = _AsType(np.random.randn(*((first_dim,) + extra_shape)), vtype)
scatter_nd_ops_test.py 36 def _AsType(v, vtype):
124 indices = _AsType(indices[:num_updates], itype)
129 updates = _AsType(np.random.randn(*(updates_shape)), vtype)
130 ref = _AsType(np.random.randn(*(ref_shape)), vtype)

Completed in 213 milliseconds