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.cc80 auto serialized_sparse_t = serialized_sparse.vec<T>(); in Compute() local
81 OP_REQUIRES_OK(context, Serialize(*input_indices, &serialized_sparse_t(0))); in Compute()
82 OP_REQUIRES_OK(context, Serialize(*input_values, &serialized_sparse_t(1))); in Compute()
83 OP_REQUIRES_OK(context, Serialize(*input_shape, &serialized_sparse_t(2))); in Compute()
202 auto serialized_sparse_t = serialized_sparse.matrix<U>(); in Compute() local
211 serialized_sparse_t.template chip<1>(0).setConstant(serialized_indices); in Compute()
217 serialized_sparse_t.template chip<1>(1).setConstant(serialized_values); in Compute()
224 serialized_sparse_t.template chip<1>(2).setConstant(serialized_shape); in Compute()
254 context, this->Serialize(output_indices, &serialized_sparse_t(b, 0))); in Compute()
256 context, this->Serialize(output_values, &serialized_sparse_t(b, 1))); in Compute()
Ddeserialize_sparse_string_op.cc78 const auto& serialized_sparse_t = serialized_sparse.vec<string>(); in Compute() local
85 serialized_sparse_t(0), serialized_sparse_t(1), in Compute()
86 serialized_sparse_t(2), dtype_, 0 /* index */, in Compute()
100 const auto& serialized_sparse_t = in Compute() local
108 serialized_sparse_t(i, 0), serialized_sparse_t(i, 1), in Compute()
109 serialized_sparse_t(i, 2), dtype_, i, &output_indices, in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc272 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in GetNextInternal() local
273 serialized_sparse_t(0) = std::move(indices); in GetNextInternal()
274 serialized_sparse_t(1) = std::move(values); in GetNextInternal()
275 serialized_sparse_t(2) = std::move(dense_shape); in GetNextInternal()
Dparse_example_dataset_op.cc317 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in MapFunc() local
318 serialized_sparse_t(0) = example_result.sparse_indices[d]; in MapFunc()
319 serialized_sparse_t(1) = example_result.sparse_values[d]; in MapFunc()
320 serialized_sparse_t(2) = example_result.sparse_shapes[d]; in MapFunc()