Searched defs:bwdParams (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.h | 214 explicit MklPoolingBwdPrimitive(const MklPoolingParams& bwdParams) in MklPoolingBwdPrimitive()
301 static MklPoolingBwdPrimitive<T>* Get(const MklPoolingParams& bwdParams) { in Get()
330 static string CreateKey(const MklPoolingParams& bwdParams) { in CreateKey()
344 MklPrimitive* GetPoolingBwd(const MklPoolingParams& bwdParams) { in GetPoolingBwd()
349 void SetPoolingBwd(const MklPoolingParams& bwdParams, MklPrimitive* op) { in SetPoolingBwd()
|
| D | mkl_relu_op.cc | 242 explicit MklEltwiseBwdPrimitive(const MklEltwiseBwdParams<T>& bwdParams) in MklEltwiseBwdPrimitive()
335 void Setup(const MklEltwiseBwdParams<T>& bwdParams) { in Setup()
383 const MklEltwiseBwdParams<T>& bwdParams) { in Get()
405 static string CreateKey(const MklEltwiseBwdParams<T>& bwdParams) { in CreateKey()
416 MklPrimitive* GetEltwiseBwd(const MklEltwiseBwdParams<T>& bwdParams) { in GetEltwiseBwd()
421 void SetEltwiseBwd(const MklEltwiseBwdParams<T>& bwdParams, in SetEltwiseBwd()
680 MklEltwiseBwdParams<T> bwdParams(src_dims, common_md, alg_kind, alpha_, in Compute() local
|
| D | mkl_fused_batch_norm_op.cc | 402 explicit MklFusedBatchNormBwdPrimitive(const MklBatchNormBwdParams& bwdParams) in MklFusedBatchNormBwdPrimitive()
519 void Setup(const MklBatchNormBwdParams& bwdParams) { in Setup()
587 const MklBatchNormBwdParams& bwdParams) { in Get()
608 static string CreateKey(const MklBatchNormBwdParams& bwdParams) { in CreateKey()
622 MklPrimitive* GetBatchNormBwd(const MklBatchNormBwdParams& bwdParams) { in GetBatchNormBwd()
627 void SetBatchNormBwd(const MklBatchNormBwdParams& bwdParams, in SetBatchNormBwd()
1220 MklBatchNormBwdParams bwdParams(src_dims, diff_dst_dims, depth_, epsilon_, in Compute() local
|
| D | mkl_pooling_ops_common.cc | 129 void MklPoolingBwdPrimitive<T>::Setup(const MklPoolingParams& bwdParams) { in Setup()
|
| D | mkl_avgpooling_op.cc | 242 MklPoolingParams bwdParams( in Compute() local
|
| D | mkl_maxpooling_op.cc | 297 MklPoolingParams bwdParams( in Compute() local
|