Home
last modified time | relevance | path

Searched defs:fwdParams (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc49 explicit MklSoftmaxPrimitive(const MklSoftmaxParams& fwdParams) in MklSoftmaxPrimitive()
114 void Setup(const MklSoftmaxParams& fwdParams) { in Setup()
146 static MklSoftmaxPrimitive<T>* Get(const MklSoftmaxParams& fwdParams) { in Get()
169 static string CreateKey(const MklSoftmaxParams& fwdParams) { in CreateKey()
180 MklPrimitive* GetSoftmaxFwd(const MklSoftmaxParams& fwdParams) { in GetSoftmaxFwd()
185 void SetSoftmaxFwd(const MklSoftmaxParams& fwdParams, MklPrimitive* op) { in SetSoftmaxFwd()
271 MklSoftmaxParams fwdParams(src_dims, src_fmt, axis); in Compute() local
Dmkl_quantize_op.cc78 const MklReorderWithScaleFwdParams& fwdParams) in MklReorderWithScalePrimitive()
127 void Setup(const MklReorderWithScaleFwdParams& fwdParams) { in Setup()
160 const MklReorderWithScaleFwdParams& fwdParams) { in Get()
183 const MklReorderWithScaleFwdParams& fwdParams) { in CreateKey()
199 const MklReorderWithScaleFwdParams& fwdParams) { in GetReorder()
205 const MklReorderWithScaleFwdParams& fwdParams) { in SetReorder()
468 MklReorderWithScaleFwdParams fwdParams(src_dims, src_md, dst_md); in Compute() local
Dmkl_pooling_ops_common.h75 explicit MklPoolingFwdPrimitive(const MklPoolingParams& fwdParams) in MklPoolingFwdPrimitive()
156 static MklPoolingFwdPrimitive<T>* Get(const MklPoolingParams& fwdParams) { in Get()
185 static string CreateKey(const MklPoolingParams& fwdParams) { in CreateKey()
200 MklPrimitive* GetPoolingFwd(const MklPoolingParams& fwdParams) { in GetPoolingFwd()
205 void SetPoolingFwd(const MklPoolingParams& fwdParams, MklPrimitive* op) { in SetPoolingFwd()
Dmkl_relu_op.cc62 explicit MklEltwiseFwdPrimitive(const MklEltwiseFwdParams<T>& fwdParams) in MklEltwiseFwdPrimitive()
136 void Setup(const MklEltwiseFwdParams<T>& fwdParams) { in Setup()
168 const MklEltwiseFwdParams<T>& fwdParams) { in Get()
193 static string CreateKey(const MklEltwiseFwdParams<T>& fwdParams) { in CreateKey()
204 MklPrimitive* GetEltwiseFwd(const MklEltwiseFwdParams<T>& fwdParams) { in GetEltwiseFwd()
209 void SetEltwiseFwd(const MklEltwiseFwdParams<T>& fwdParams, in SetEltwiseFwd()
476 MklEltwiseFwdParams<T> fwdParams(src_dims, src_md, alg_kind, alpha_, in Compute() local
Dmkl_fused_batch_norm_op.cc65 explicit MklFusedBatchNormFwdPrimitive(const MklBatchNormFwdParams& fwdParams) in MklFusedBatchNormFwdPrimitive()
185 void Setup(const MklBatchNormFwdParams& fwdParams) { in Setup()
329 const MklBatchNormFwdParams& fwdParams) { in Get()
351 static string CreateKey(const MklBatchNormFwdParams& fwdParams) { in CreateKey()
365 MklPrimitive* GetBatchNormFwd(const MklBatchNormFwdParams& fwdParams) { in GetBatchNormFwd()
370 void SetBatchNormFwd(const MklBatchNormFwdParams& fwdParams, in SetBatchNormFwd()
776 MklBatchNormFwdParams fwdParams(src_dims, depth_, epsilon_, is_training_, in Compute() local
Dmkl_pooling_ops_common.cc31 void MklPoolingFwdPrimitive<T>::Setup(const MklPoolingParams& fwdParams) { in Setup()
Dmkl_avgpooling_op.cc113 MklPoolingParams fwdParams( in Compute() local
Dmkl_maxpooling_op.cc139 MklPoolingParams fwdParams( in Compute() local