Home
last modified time | relevance | path

Searched refs:sparse_slice (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_slice_op_test.py87 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [2, 6])
88 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [2, 0], [3, 7])
105 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 3])
106 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 3], [5, 2])
107 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 5], [5, 2])
123 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 2])
124 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 2], [5, 2])
125 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 4], [5, 2])
126 sp_tensor3 = sparse_ops.sparse_slice(sp_input, [0, 6], [5, 2])
147 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [3, 7])
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseSlice.pbtxt59 sparse_slice([0, 0], [2, 4]) = shape = [2, 4]
63 sparse_slice([0, 4], [2, 3]) = shape = [2, 3]
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1152 auto work = [counter, transpose](SparseSlice<TL>* sparse_slice, in CreateSparseSlices()
1156 sparse_slice->template Initialize<true>(*slice, col_offset); in CreateSparseSlices()
1158 sparse_slice->template Initialize<false>(*slice, col_offset); in CreateSparseSlices()
1179 auto* sparse_slice = in CreateSparseSlices() local
1181 (*mat_slices)[i][j] = sparse_slice; in CreateSparseSlices()
1183 [=]() { work(sparse_slice, slice, slice_num_cols * j); }); in CreateSparseSlices()
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py782 sp_mat = sparse_ops.sparse_slice(sp_mat_full,
844 sp_mat = sparse_ops.sparse_slice(sp_mat_full, (1, 0),
Dsparse_ops.py913 def sparse_slice(sp_input, start, size, name=None): function
949 output_indices, output_values, output_shape = gen_sparse_ops.sparse_slice(
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py998 weighted_column = sparse_ops.sparse_slice(
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py3027 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
Dfeature_column_v2.py4193 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2144 name: "sparse_slice"