Home
last modified time | relevance | path

Searched refs:sparse_slice (Results 1 – 10 of 10) 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])
104 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 3])
105 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 3], [5, 2])
106 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 5], [5, 2])
122 sp_tensor0 = sparse_ops.sparse_slice(sp_input, [0, 0], [5, 2])
123 sp_tensor1 = sparse_ops.sparse_slice(sp_input, [0, 2], [5, 2])
124 sp_tensor2 = sparse_ops.sparse_slice(sp_input, [0, 4], [5, 2])
125 sp_tensor3 = sparse_ops.sparse_slice(sp_input, [0, 6], [5, 2])
146 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/
Darray_grad.py1031 sp_mat = sparse_ops.sparse_slice(sp_mat_full, (1, 0),
1093 sp_mat = sparse_ops.sparse_slice(sp_mat_full, (1, 0),
Dsparse_ops.py1128 def sparse_slice(sp_input, start, size, name=None): function
1164 output_indices, output_values, output_shape = gen_sparse_ops.sparse_slice(
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2.py867 sparse_indices = sparse_ops.sparse_slice(sparse_indices, [0, 0],
Dtpu_embedding_v2.py1611 truncated_inp = sparse_ops.sparse_slice(inp, start=[0, 0],
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py3136 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
Dfeature_column_v2.py4094 weighted_column = sparse_ops.sparse_slice(weighted_column, [0, 0],
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2228 name: "sparse_slice"