| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_split_op.cc | 107 auto sp_shape = sp_shape_t->vec<int64>(); in Compute() local
108 sp_shape(0) = batch_size; in Compute()
109 sp_shape(1) = max_num_entries; in Compute()
|
| D | sparse_dense_binary_op_shared_test.cc | 212 Node* sp_shape, Node* dense) { in SparseMatCMulDenseMat() argument
218 .Input(sp_shape) in SparseMatCMulDenseMat()
|
| D | sparse_reduce_op.cc | 85 auto sp_shape = sp.shape(); in SparseTensorReduceHelper() local
95 out_dim_sizes.push_back(sp_shape[d]); in SparseTensorReduceHelper()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 112 sp_shape = op.inputs[2]
113 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
115 scale = sp_shape // math_ops.to_int64(output_shape_kept_dims)
|
| D | array_grad.py | 786 sp_shape = (rows_in * cols_in, rows_out * cols_out * ksize_r * ksize_c)
790 array_ops.ones((len(idx),), dtype=ops.dtypes.float32), sp_shape)
|
| D | math_ops.py | 995 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): argument
998 [sp_indices, sp_values, sp_shape, y]) as name:
1014 sp_indices, sp_values, sp_shape, y, name=name)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 616 sp_shape = array_ops.identity(sp.dense_shape)
617 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
620 [sp_indices, sp_values, sp_shape], {
633 [sp_indices, sp_values, sp_shape], {
662 sp_shape = array_ops.identity(sp.dense_shape)
663 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
666 [sp_indices, sp_values, sp_shape], {
674 [sp_indices, sp_values, sp_shape], {
697 sp_shape = array_ops.identity(sp.dense_shape)
698 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseDenseCwiseDiv.pbtxt | 17 name: "sp_shape"
|
| D | api_def_SparseDenseCwiseAdd.pbtxt | 17 name: "sp_shape"
|
| D | api_def_SparseDenseCwiseMul.pbtxt | 17 name: "sp_shape"
|
| D | api_def_SparseSoftmax.pbtxt | 17 name: "sp_shape"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 730 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes):
731 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1
772 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes):
773 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1
|
| D | lookup_ops_test.py | 162 sp_shape = [2, 2]
166 constant_op.constant(sp_shape, dtypes.int64))
173 self.assertAllEqual(sp_shape, out_shape)
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops_test.py | 168 sp_shape = [2, 2]
172 constant_op.constant(sp_shape, dtypes.int64))
179 self.assertAllEqual(sp_shape, out_shape)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 5771 func SparseSoftmax(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Output) (ou…
5778 sp_indices, sp_values, sp_shape,
8171 func SparseDenseCwiseMul(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
8178 sp_indices, sp_values, sp_shape, dense,
11366 func SparseDenseCwiseDiv(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
11373 sp_indices, sp_values, sp_shape, dense,
11542 func SparseDenseCwiseAdd(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
11549 sp_indices, sp_values, sp_shape, dense,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 56892 name: "sp_shape"
56937 name: "sp_shape"
56984 name: "sp_shape"
57032 name: "sp_shape"
57080 name: "sp_shape"
57125 name: "sp_shape"
57172 name: "sp_shape"
57220 name: "sp_shape"
57268 name: "sp_shape"
57313 name: "sp_shape"
[all …]
|
| D | ops_history.v0.pbtxt | 33676 name: "sp_shape"
33721 name: "sp_shape"
33766 name: "sp_shape"
34539 name: "sp_shape"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25865 name: "sp_shape"
25913 name: "sp_shape"
25961 name: "sp_shape"
26870 name: "sp_shape"
|