Home
last modified time | relevance | path

Searched refs:serialized_sparse_t (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dserialize_sparse_op.cc82 auto serialized_sparse_t = serialized_sparse.vec<T>(); in Compute() local
83 OP_REQUIRES_OK(context, Serialize(*input_indices, &serialized_sparse_t(0))); in Compute()
84 OP_REQUIRES_OK(context, Serialize(*input_values, &serialized_sparse_t(1))); in Compute()
85 OP_REQUIRES_OK(context, Serialize(*input_shape, &serialized_sparse_t(2))); in Compute()
152 auto serialized_sparse_t = serialized_sparse->matrix<tstring>(); in operator ()() local
166 serialize(Tensor(DT_INT64, {0, rank - 1}), &serialized_sparse_t(b, 0)); in operator ()()
168 &serialized_sparse_t(b, 1)); in operator ()()
169 serialized_sparse_t(b, 2) = serialized_shape; in operator ()()
206 serialize(output_indices, &serialized_sparse_t(b, 0)); in operator ()()
207 serialize(output_values, &serialized_sparse_t(b, 1)); in operator ()()
[all …]
Ddeserialize_sparse_string_op.cc80 const auto& serialized_sparse_t = serialized_sparse.vec<tstring>(); in Compute() local
87 serialized_sparse_t(0), serialized_sparse_t(1), in Compute()
88 serialized_sparse_t(2), dtype_, 0 /* index */, in Compute()
102 const auto& serialized_sparse_t = in Compute() local
110 serialized_sparse_t(i, 0), serialized_sparse_t(i, 1), in Compute()
111 serialized_sparse_t(i, 2), dtype_, i, &output_indices, in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc280 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in GetNextInternal() local
281 serialized_sparse_t(0) = std::move(indices); in GetNextInternal()
282 serialized_sparse_t(1) = std::move(values); in GetNextInternal()
283 serialized_sparse_t(2) = std::move(dense_shape); in GetNextInternal()
Dparse_example_dataset_op.cc672 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in ParseExample() local
673 serialized_sparse_t(0) = example_result.sparse_indices[d]; in ParseExample()
674 serialized_sparse_t(1) = example_result.sparse_values[d]; in ParseExample()
675 serialized_sparse_t(2) = example_result.sparse_shapes[d]; in ParseExample()