Home
last modified time | relevance | path

Searched refs:sp_shape (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSparseDenseCwiseAdd.pbtxt12 name: "sp_shape"
57 name: "sp_shape"
104 name: "sp_shape"
152 name: "sp_shape"
DSparseDenseCwiseDiv.pbtxt12 name: "sp_shape"
57 name: "sp_shape"
104 name: "sp_shape"
152 name: "sp_shape"
DSparseDenseCwiseMul.pbtxt12 name: "sp_shape"
57 name: "sp_shape"
104 name: "sp_shape"
152 name: "sp_shape"
DSparseSoftmax.pbtxt12 name: "sp_shape"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSparseDenseCwiseMul.pbtxt12 name: "sp_shape"
57 name: "sp_shape"
104 name: "sp_shape"
152 name: "sp_shape"
DSparseDenseCwiseAdd.pbtxt12 name: "sp_shape"
57 name: "sp_shape"
104 name: "sp_shape"
152 name: "sp_shape"
DSparseDenseCwiseDiv.pbtxt12 name: "sp_shape"
57 name: "sp_shape"
104 name: "sp_shape"
152 name: "sp_shape"
DSparseSoftmax.pbtxt12 name: "sp_shape"
/external/tensorflow/tensorflow/core/kernels/
Dstring_split_op.cc224 auto sp_shape = sp_shape_t->vec<int64_t>(); 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_t>(); in Compute() local
295 sp_shape(0) = batch_size; in Compute()
296 sp_shape(1) = max_num_entries; in Compute()
Dsparse_dense_binary_op_shared_test.cc214 Node* sp_shape, Node* dense) { in SparseMatCMulDenseMat() argument
220 .Input(sp_shape) in SparseMatCMulDenseMat()
Dsparse_reduce_op.cc87 auto sp_shape = sp.shape(); in SparseTensorReduceHelper() local
97 out_dim_sizes.push_back(sp_shape[d]); in SparseTensorReduceHelper()
/external/tensorflow/tensorflow/python/ops/
Dsparse_grad.py103 sp_shape = op.inputs[2]
104 output_shape_kept_dims = math_ops.reduced_shape(sp_shape, op.inputs[3])
106 scale = sp_shape // math_ops.cast(output_shape_kept_dims, dtypes.int64)
Darray_grad.py1053 sp_shape = (input_indices_num, output_indices_num)
1055 idx_map, array_ops.ones([output_indices_num], dtype=grad.dtype), sp_shape)
1115 sp_shape = (input_indices_num, output_indices_num)
1117 idx_map, array_ops.ones([output_indices_num], dtype=grad.dtype), sp_shape)
Dmath_ops.py1486 def _sparse_dense_truediv(sp_indices, sp_values, sp_shape, y, name=None): argument
1489 [sp_indices, sp_values, sp_shape, y]) as name:
1507 sp_indices, sp_values, sp_shape, y, name=name)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseDenseCwiseDiv.pbtxt17 name: "sp_shape"
Dapi_def_SparseDenseCwiseAdd.pbtxt17 name: "sp_shape"
Dapi_def_SparseDenseCwiseMul.pbtxt17 name: "sp_shape"
Dapi_def_SparseSoftmax.pbtxt17 name: "sp_shape"
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-110.md17 sp_shape=np.array([647.]),
Dtfsa-2021-056.md20 sp_indices=indices, sp_values=values, sp_shape=shape, dense=dense)
Dtfsa-2022-011.md16 sp_shape=np.array([92233720368., 92233720368]),
/external/tensorflow/tensorflow/python/client/
Dsession_test.py718 sp_shape = array_ops.identity(sp.dense_shape)
719 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
722 [sp_indices, sp_values, sp_shape], {
735 [sp_indices, sp_values, sp_shape], {
764 sp_shape = array_ops.identity(sp.dense_shape)
765 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
768 [sp_indices, sp_values, sp_shape], {
776 [sp_indices, sp_values, sp_shape], {
799 sp_shape = array_ops.identity(sp.dense_shape)
800 sp2 = sparse_tensor.SparseTensor(sp_indices, sp_values, sp_shape)
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_ops_test.py916 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes):
917 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1
959 for sp_shape, dense_shape in zip(sp_shapes, dense_shapes):
960 sp_vals_np = np.random.rand(*sp_shape).astype(dtype) + 1
1052 sp_shape=[2**32, 2**31],
1061 sp_indices=[[1, 1]], sp_values=[2.0], sp_shape=[-1, 1], name=None)
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dlookup_ops_test.py266 sp_shape = [2, 2]
270 constant_op.constant(sp_shape, dtypes.int64))
277 self.assertAllEqual(sp_shape, out_shape)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4353 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
4357 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
4361 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'dense\', \'name\'], varargs=None, k…
4489 …argspec: "args=[\'sp_indices\', \'sp_values\', \'sp_shape\', \'name\'], varargs=None, keywords=Non…

12