Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dexample_proto_fast_parsing.cc25 const std::vector<std::vector<SparseBuffer>>& sparse_buffers, size_t d, in CountSparseFeatures()
39 void CopySparseBufferToTensor(DataType dtype, size_t offset, SparseBuffer* src, in CopySparseBufferToTensor()
156 const SmallVector<int64>& GetListFromBuffer<int64>(const SparseBuffer& buffer) { in GetListFromBuffer()
160 const SmallVector<float>& GetListFromBuffer<float>(const SparseBuffer& buffer) { in GetListFromBuffer()
165 const SparseBuffer& buffer) { in GetListFromBuffer()
Dexample_proto_fast_parsing.h375 struct SparseBuffer { struct
404 const SmallVector<T>& GetListFromBuffer(const SparseBuffer& buffer); argument
407 const SmallVector<int64>& GetListFromBuffer<int64>(const SparseBuffer& buffer);
410 const SmallVector<float>& GetListFromBuffer<float>(const SparseBuffer& buffer);
414 const SparseBuffer& buffer);
424 const std::vector<std::vector<SparseBuffer>>& sparse_buffers, size_t d,
427 void CopySparseBufferToTensor(DataType dtype, size_t offset, SparseBuffer* src,
Dparse_example.cc57 using tf::example::SparseBuffer;
75 std::vector<SparseBuffer>& varlen_dense_buffers, in FillAndCopyVarLen()
86 const SparseBuffer& buffer = varlen_dense_buffers[d]; in FillAndCopyVarLen()
123 std::vector<SparseBuffer>* output_varlen_dense, in FastParseSerializedExample()
124 std::vector<SparseBuffer>* output_sparse, in FastParseSerializedExample()
228 SparseBuffer& out = (*output_varlen_dense)[d]; in FastParseSerializedExample()
296 SparseBuffer& out = (*output_sparse)[d]; in FastParseSerializedExample()
374 SparseBuffer& out = (*output_varlen_dense)[d]; in FastParseSerializedExample()
382 SparseBuffer& out = (*output_sparse)[d]; in FastParseSerializedExample()
391 void CountSparseFeatures(const SparseBuffer& sparse_buffer, in CountSparseFeatures()
[all …]
/external/tensorflow/tensorflow/core/util/
Dexample_proto_fast_parsing.cc529 struct SparseBuffer { struct
576 std::vector<SparseBuffer>* output_varlen_dense, in FastParseSerializedExample()
577 std::vector<SparseBuffer>* output_sparse, in FastParseSerializedExample()
578 std::vector<SparseBuffer>* output_ragged, in FastParseSerializedExample()
713 SparseBuffer& out = (*output_varlen_dense)[d]; in FastParseSerializedExample()
804 SparseBuffer& out = is_ragged ? (*output_ragged)[d] : (*output_sparse)[d]; in FastParseSerializedExample()
905 SparseBuffer& out = (*output_varlen_dense)[d]; in FastParseSerializedExample()
914 SparseBuffer& out = (*output_sparse)[d]; in FastParseSerializedExample()
923 SparseBuffer& out = (*output_ragged)[d]; in FastParseSerializedExample()
971 const SmallVector<T>& GetListFromBuffer(const SparseBuffer& buffer);
[all …]