Searched refs:sparse_slice (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | sparse_slice_op_test.py | 87 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/ |
D | api_def_SparseSlice.pbtxt | 59 sparse_slice([0, 0], [2, 4]) = shape = [2, 4] 63 sparse_slice([0, 4], [2, 3]) = shape = [2, 3]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_matmul_op.cc | 1152 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/ |
D | array_grad.py | 782 sp_mat = sparse_ops.sparse_slice(sp_mat_full, 844 sp_mat = sparse_ops.sparse_slice(sp_mat_full, (1, 0),
|
D | sparse_ops.py | 913 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/ |
D | feature_column.py | 998 weighted_column = sparse_ops.sparse_slice(
|
/external/tensorflow/tensorflow/python/feature_column/ |
D | feature_column.py | 3027 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
|
D | feature_column_v2.py | 4193 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.pbtxt | 2144 name: "sparse_slice"
|