mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
2e4c89eba9
Summary: 1) Adding MKL/AVX2 based implementation into perfkernels. This implementation is similar to caffe2/operators/batch_box_cox_op.cc 2) Migrating batch_box_cox_op of caffe2 use this implementation Test Plan: CI Differential Revision: D40208074 Pull Request resolved: https://github.com/pytorch/pytorch/pull/86569 Approved by: https://github.com/hyuen
36 lines
789 B
C++
36 lines
789 B
C++
// Impmenets BoxCox operator for CPU
|
|
#pragma once
|
|
#include <cstdint>
|
|
|
|
namespace caffe2 {
|
|
|
|
template <typename T>
|
|
void compute_batch_box_cox(
|
|
std::size_t N,
|
|
std::size_t D,
|
|
std::size_t block_size,
|
|
const T* self_data,
|
|
const T* lambda1_data,
|
|
const T* lambda2_data,
|
|
T* output_data);
|
|
|
|
extern template void compute_batch_box_cox<float>(
|
|
std::size_t N,
|
|
std::size_t D,
|
|
std::size_t block_size,
|
|
const float* data,
|
|
const float* lambda1_data,
|
|
const float* lambda2_data,
|
|
float* output_data);
|
|
|
|
extern template void compute_batch_box_cox<double>(
|
|
std::size_t N,
|
|
std::size_t D,
|
|
std::size_t block_size,
|
|
const double* data,
|
|
const double* lambda1_data,
|
|
const double* lambda2_data,
|
|
double* output_data);
|
|
|
|
} // namespace caffe2
|