Caffe2 - C++ API
A deep learning, cross platform ML framework
fused_8bit_rowwise_embedding_lookup.h
1 #pragma once
2 
3 #include "caffe2/core/common.h"
4 
5 namespace caffe2 {
6 
39 template <
40  typename IndexType,
41  typename InType,
42  typename OutType,
43  bool IS_WEIGHT_POSITIONAL = false>
44 void Fused8BitRowwiseEmbeddingLookup(
45  const TIndex block_size,
46  const TIndex output_size,
47  const TIndex index_size,
48  const TIndex 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
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition: convert_encoded_to_raw_leveldb.cc:47
caffe2::Fused8BitRowwiseEmbeddingLookup
void Fused8BitRowwiseEmbeddingLookup(const TIndex block_size, const TIndex output_size, const TIndex index_size, const TIndex data_size, const InType *input, const IndexType *indices, const int *lengths, const float *weights, bool normalize_by_lengths, OutType *out)
Embedding lookup with reduction.
Generated on Thu Apr 19 2018 13:03:57 for Caffe2 - C++ API by   doxygen 1.8.11