Caffe2 - C++ API
A deep learning, cross platform ML framework
fused_8bit_rowwise_embedding_lookup.h
1 #pragma once
2 
3 #include <cstdint>
4 
5 namespace caffe2 {
6 
39 template <
40  typename IndexType,
41  typename InType,
42  typename OutType,
43  bool IS_WEIGHT_POSITIONAL = false>
45  const std::int64_t block_size,
46  const std::int64_t output_size,
47  const std::int64_t index_size,
48  const std::int64_t data_size,
49  const InType* input,
50  const IndexType* indices,
51  const int* lengths,
52  const float* weights, // optional, can be null for non-weighted sum
53  bool normalize_by_lengths,
54  OutType* out);
55 } // namespace caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: blob.h:13
void Fused8BitRowwiseEmbeddingLookup(const std::int64_t block_size, const std::int64_t output_size, const std::int64_t index_size, const std::int64_t data_size, const InType *input, const IndexType *indices, const int *lengths, const float *weights, bool normalize_by_lengths, OutType *out)
Embedding lookup with reduction.