Home
last modified time | relevance | path

Searched refs:bincount (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dbincount_op_test.py41 self.evaluate(bincount_ops.bincount([], minlength=5)),
44 self.evaluate(bincount_ops.bincount([], minlength=1)), [0])
46 self.evaluate(bincount_ops.bincount([], minlength=0)), [])
49 bincount_ops.bincount([], minlength=0, dtype=np.float32)).dtype,
53 bincount_ops.bincount([], minlength=3, dtype=np.float64)).dtype,
59 self.evaluate(bincount_ops.bincount([1, 1, 1, 2, 2, 3])),
63 self.evaluate(bincount_ops.bincount(arr)), [0, 5, 4, 3, 2, 1])
66 self.evaluate(bincount_ops.bincount(arr)), [6, 5, 4, 3, 2, 1])
68 self.assertAllEqual(self.evaluate(bincount_ops.bincount([])), [])
69 self.assertAllEqual(self.evaluate(bincount_ops.bincount([0, 0, 0])), [3])
[all …]
/external/tensorflow/tensorflow/python/ops/
Dbincount_ops_test.py534 np_out = np.bincount(inp_vals, minlength=size)
536 np_out, self.evaluate(bincount_ops.bincount(sparse_inp, axis=0)))
557 np_out = np.bincount(inp_vals, minlength=size, weights=weight_vals)
560 self.evaluate(bincount_ops.bincount(
582 self.evaluate(bincount_ops.bincount(sparse_inp, binary_output=True)))
597 [np.bincount(inp[j, :], minlength=size) for j in range(num_rows)],
608 np_out, self.evaluate(bincount_ops.bincount(arr=inp_sparse, axis=-1)))
623 np.where(np.bincount(inp[j, :], minlength=size) > 0, 1, 0)
638 bincount_ops.bincount(arr=inp_sparse, axis=-1, binary_output=True)))
657 self.evaluate(bincount_ops.bincount(arr=x, axis=-1)))
[all …]
Dbincount_ops.py36 def bincount(arr, function
141 return gen_math_ops.bincount(arr, output_size, weights)
251 return bincount(arr, weights, minlength, maxlength, dtype)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Drejection_resample_test.py131 bincount = np.bincount(
135 self.assertAllClose(target_dist, bincount, atol=1e-2)
/external/tensorflow/tensorflow/python/ops/ragged/
Dsegment_id_ops.py124 row_lengths = bincount_ops.bincount(
Drow_partition.py284 row_lengths = bincount_ops.bincount(
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_encoding.py212 result = bincount_ops.bincount(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt12 name: "bincount"
Dtensorflow.math.pbtxt84 name: "bincount"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt16 name: "bincount"
Dtensorflow.math.pbtxt84 name: "bincount"
Dtensorflow.pbtxt932 name: "bincount"
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py77 nz = np.bincount(a_mats.nonzero()[0], minlength=a_mats.shape[0])
/external/tensorflow/
DRELEASE.md3975 * Added a `tf.bincount` function.