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.cc1151 auto work = [counter, transpose](SparseSlice<TL>* sparse_slice, in CreateSparseSlices()
1155 sparse_slice->template Initialize<true>(*slice, col_offset); in CreateSparseSlices()
1157 sparse_slice->template Initialize<false>(*slice, col_offset); in CreateSparseSlices()
1178 auto* sparse_slice = in CreateSparseSlices() local
1180 (*mat_slices)[i][j] = sparse_slice; in CreateSparseSlices()
1182 [=]() { work(sparse_slice, slice, slice_num_cols * j); }); in CreateSparseSlices()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py964 def sparse_slice(sp_input, start, size, name=None): function
1000 output_indices, output_values, output_shape = gen_sparse_ops.sparse_slice(
Darray_grad.py1003 sp_mat = sparse_ops.sparse_slice(sp_mat_full, (1, 0),
1065 sp_mat = sparse_ops.sparse_slice(sp_mat_full, (1, 0),
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2.py832 sparse_indices = sparse_ops.sparse_slice(sparse_indices, [0, 0],
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py3111 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
Dfeature_column_v2.py4367 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2220 name: "sparse_slice"