Home
last modified time | relevance | path

Searched refs:sparse_indices (Results 1 – 25 of 74) sorted by relevance

123

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseToDense.pbtxt4 name: "sparse_indices"
6 0-D, 1-D, or 2-D. `sparse_indices[i]` contains the complete
19 1-D. Values corresponding to each row of `sparse_indices`,
27 `sparse_indices`.
48 # If sparse_indices is scalar
49 dense[i] = (i == sparse_indices ? sparse_values : default_value)
51 # If sparse_indices is a vector, then for each i
52 dense[sparse_indices[i]] = sparse_values[i]
54 # If sparse_indices is an n by d matrix, then for each i in [0, n)
55 dense[sparse_indices[i][0], ..., sparse_indices[i][d-1]] = sparse_values[i]
Dapi_def_AddManySparseToTensorsMap.pbtxt4 name: "sparse_indices"
7 `sparse_indices[:, 0]` must be ordered values in `[0, N)`.
45 A `SparseTensor` of rank `R` is represented by three tensors: `sparse_indices`,
48 ```sparse_indices.shape[1] == sparse_shape.shape[0] == R```
51 having a first `sparse_indices` column taking values between `[0, N)`, where
Dapi_def_RaggedTensorToSparse.pbtxt13 name: "sparse_indices"
39 output=SparseTensor(indices=sparse_indices, values=sparse_values,
Dapi_def_AddSparseToTensorsMap.pbtxt4 name: "sparse_indices"
43 A `SparseTensor` is represented by three tensors: `sparse_indices`,
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSerializeManySparse.pbtxt4 name: "sparse_indices"
27 name: "sparse_indices"
DSerializeSparse.pbtxt4 name: "sparse_indices"
27 name: "sparse_indices"
DDeserializeSparse.pbtxt8 name: "sparse_indices"
31 name: "sparse_indices"
DRaggedTensorToSparse.pbtxt13 name: "sparse_indices"
47 name: "sparse_indices"
DDeserializeManySparse.pbtxt8 name: "sparse_indices"
DSdcaOptimizer.pbtxt32 name: "sparse_indices"
150 name: "sparse_indices"
DTakeManySparseFromTensorsMap.pbtxt8 name: "sparse_indices"
DAddSparseToTensorsMap.pbtxt4 name: "sparse_indices"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDeserializeSparse.pbtxt8 name: "sparse_indices"
31 name: "sparse_indices"
DSerializeManySparse.pbtxt4 name: "sparse_indices"
27 name: "sparse_indices"
DSerializeSparse.pbtxt4 name: "sparse_indices"
27 name: "sparse_indices"
DRaggedTensorToSparse.pbtxt13 name: "sparse_indices"
47 name: "sparse_indices"
DDeserializeManySparse.pbtxt8 name: "sparse_indices"
DSdcaOptimizer.pbtxt32 name: "sparse_indices"
150 name: "sparse_indices"
DAddSparseToTensorsMap.pbtxt4 name: "sparse_indices"
DAddManySparseToTensorsMap.pbtxt4 name: "sparse_indices"
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_sparse_kernel.cc71 auto sparse_indices = sparse_indices_out->tensor<int64, 2>(); in Compute() local
107 sparse_indices(next_index, dim++) = index; in Compute()
109 sparse_indices(next_index, dim++) = i; // index_middle in Compute()
111 sparse_indices(next_index, dim++) = index; in Compute()
Dsparse_to_dense_op_test.cc216 Tensor sparse_indices(DT_INT64, TensorShape({N, NDIM})); in BM_SparseToDense() local
224 auto sparse_indices_t = sparse_indices.matrix<int64>(); in BM_SparseToDense()
231 {&sparse_indices, &output_shape, &sparse_values, &default_value}) { in BM_SparseToDense()
Dsdca_ops_test.cc160 std::vector<NodeBuilder::NodeOut> sparse_indices; in GetGraphs() local
163 sparse_indices.push_back( in GetGraphs()
220 .Input(sparse_indices) in GetGraphs()
/external/tensorflow/tensorflow/compiler/tests/
Dsparse_to_dense_op_test.py31 def _SparseToDense(sparse_indices, argument
37 feed_dict = {feed_sparse_indices: sparse_indices}
/external/tensorflow/tensorflow/python/ops/
Dsparse_grad.py317 sparse_indices, output_shape, _, _ = op.inputs
319 sparse_values_grad = array_ops.gather_nd(grad, sparse_indices)
323 array_ops.zeros_like(sparse_indices),

123