THCCWeighting.c

#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/THCCWeighting.c"
#else

void THCCTensor_(weightingForward)(THCTensor *self, THCTensor *src, THCTensor *weight,
                                   THCTensor *basis, THCudaLongTensor *weightIndex) {
  THCTensor_(weightingForward)(state, self, src, weight, basis, weightIndex);
}

void THCCTensor_(weightingBackwardSrc)(THCTensor *self, THCTensor *gradOutput, THCTensor *weight,
                                       THCTensor *basis, THCudaLongTensor *weightIndex) {
  THCTensor_(weightingBackwardSrc)(state, self, gradOutput, weight, basis, weightIndex);
}

void THCCTensor_(weightingBackwardWeight)(THCTensor *self, THCTensor *gradOutput, THCTensor *src,
                                          THCTensor *basis, THCudaLongTensor *weightIndex) {
  THCTensor_(weightingBackwardWeight)(state, self, gradOutput, src, basis, weightIndex);
}

void THCCTensor_(weightingBackwardBasis)(THCTensor *self, THCTensor *gradOutput, THCTensor *src,
                                         THCTensor *weight, THCudaLongTensor *weightIndex) {
  THCTensor_(weightingBackwardBasis)(state, self, gradOutput, src, weight, weightIndex);
}

#endif  // THC_GENERIC_FILE