| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_split_op.cc | 224 auto sp_shape = sp_shape_t->vec<int64>(); in Compute() local
225 sp_shape(0) = batch_size; in Compute()
226 sp_shape(1) = max_num_entries; in Compute()
294 auto sp_shape = sp_shape_t->vec<int64>(); in Compute() local
295 sp_shape(0) = batch_size; in Compute()
296 sp_shape(1) = max_num_entries; in Compute()
|
| D | sparse_dense_binary_op_shared_test.cc | 213 Node* sp_shape, Node* dense) { in SparseMatCMulDenseMat() argument
219 .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 | 110 sp_shape = op.inputs[2]
111 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
113 scale = sp_shape // math_ops.cast(output_shape_kept_dims, dtypes.int64)
|
| D | array_grad.py | 776 sp_shape = (input_indices_num, output_indices_num)
780 sp_shape)
840 sp_shape = (input_indices_num, output_indices_num)
842 idx_map, array_ops.ones([output_indices_num], dtype=grad.dtype), sp_shape)
|
| D | math_ops.py | 934 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): argument
937 [sp_indices, sp_values, sp_shape, y]) as name:
953 sp_indices, sp_values, sp_shape, y, name=name)
|
| /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/client/ |
| D | session_test.py | 701 sp_shape = array_ops.identity(sp.dense_shape)
702 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
705 [sp_indices, sp_values, sp_shape], {
718 [sp_indices, sp_values, sp_shape], {
747 sp_shape = array_ops.identity(sp.dense_shape)
748 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
751 [sp_indices, sp_values, sp_shape], {
759 [sp_indices, sp_values, sp_shape], {
782 sp_shape = array_ops.identity(sp.dense_shape)
783 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 806 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes):
807 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1
849 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes):
850 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1
|
| D | lookup_ops_test.py | 192 sp_shape = [2, 2]
196 constant_op.constant(sp_shape, dtypes.int64))
203 self.assertAllEqual(sp_shape, out_shape)
|
| /external/tensorflow/tensorflow/contrib/lookup/ |
| D | lookup_ops_test.py | 172 sp_shape = [2, 2]
176 constant_op.constant(sp_shape, dtypes.int64))
183 self.assertAllEqual(sp_shape, out_shape)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3477 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3481 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3485 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3565 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'name\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3477 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3481 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3485 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
3565 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'name\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12839 func SparseDenseCwiseMul(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
12846 sp_indices, sp_values, sp_shape, dense,
14286 func SparseDenseCwiseAdd(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
14293 sp_indices, sp_values, sp_shape, dense,
17269 func SparseSoftmax(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Output) (ou…
17276 sp_indices, sp_values, sp_shape,
23931 func SparseDenseCwiseDiv(scope *Scope, sp_indices tf.Output, sp_values tf.Output, sp_shape tf.Outpu…
23938 sp_indices, sp_values, sp_shape, dense,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 75553 name: "sp_shape"
75598 name: "sp_shape"
75645 name: "sp_shape"
75693 name: "sp_shape"
75741 name: "sp_shape"
75786 name: "sp_shape"
75833 name: "sp_shape"
75881 name: "sp_shape"
75929 name: "sp_shape"
75974 name: "sp_shape"
[all …]
|
| D | ops_history.v1.pbtxt | 76180 name: "sp_shape"
76225 name: "sp_shape"
76272 name: "sp_shape"
76320 name: "sp_shape"
76368 name: "sp_shape"
76413 name: "sp_shape"
76460 name: "sp_shape"
76508 name: "sp_shape"
76556 name: "sp_shape"
76601 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 | 36377 name: "sp_shape"
36425 name: "sp_shape"
36473 name: "sp_shape"
37430 name: "sp_shape"
|