Searched refs:MklPrimitive (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 47 class MklSoftmaxPrimitive : public MklPrimitive {
50 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklSoftmaxPrimitive()
180 MklPrimitive* GetSoftmaxFwd(const MklSoftmaxParams& fwdParams) { in GetSoftmaxFwd()
185 void SetSoftmaxFwd(const MklSoftmaxParams& fwdParams, MklPrimitive* op) { in SetSoftmaxFwd()
|
| D | mkl_pooling_ops_common.h | 73 class MklPoolingFwdPrimitive : public MklPrimitive {
76 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklPoolingFwdPrimitive()
200 MklPrimitive* GetPoolingFwd(const MklPoolingParams& fwdParams) { in GetPoolingFwd()
205 void SetPoolingFwd(const MklPoolingParams& fwdParams, MklPrimitive* op) { in SetPoolingFwd()
212 class MklPoolingBwdPrimitive : public MklPrimitive {
215 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklPoolingBwdPrimitive()
344 MklPrimitive* GetPoolingBwd(const MklPoolingParams& bwdParams) { in GetPoolingBwd()
349 void SetPoolingBwd(const MklPoolingParams& bwdParams, MklPrimitive* op) { in SetPoolingBwd()
|
| D | mkl_matmul_ops_common.h | 83 class MklDnnMatMulFwdPrimitive : public MklPrimitive {
87 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklDnnMatMulFwdPrimitive()
368 MklPrimitive* GetMklDnnMatMulFwd( in GetMklDnnMatMulFwd()
375 MklPrimitive* op) { in SetMklDnnMatMulFwd()
531 class MklMatMulPrimitive : public MklPrimitive {
534 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklMatMulPrimitive()
683 MklPrimitive* GetMklMatMul(const MklMatMulParams& params) { in GetMklMatMul()
688 void SetMklMatMul(const MklMatMulParams& params, MklPrimitive* op) { in SetMklMatMul()
|
| D | mkl_quantize_op.cc | 75 class MklReorderWithScalePrimitive : public MklPrimitive {
79 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklReorderWithScalePrimitive()
198 MklPrimitive* GetReorder(const memory* from, const memory* to, in GetReorder()
204 void SetReorder(const memory* from, const memory* to, MklPrimitive* op, in SetReorder()
|
| D | mkl_slice_op.cc | 177 class MklSlicePrimitive : public MklPrimitive {
180 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklSlicePrimitive()
300 MklPrimitive* GetReorder(const MklSliceParams& sliceParams) { in GetReorder()
305 void SetReorder(const MklSliceParams& sliceParams, MklPrimitive* op) { in SetReorder()
|
| D | mkl_conv_grad_input_ops.cc | 88 class MklConvBwdInputPrimitive : public MklPrimitive {
92 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklConvBwdInputPrimitive()
291 MklPrimitive* GetConvBwdInput(const MklConvBwdInputParams& convBwdInputDims) { in GetConvBwdInput()
297 MklPrimitive* op) { in SetConvBwdInput()
|
| D | mkl_relu_op.cc | 60 class MklEltwiseFwdPrimitive : public MklPrimitive {
63 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklEltwiseFwdPrimitive()
204 MklPrimitive* GetEltwiseFwd(const MklEltwiseFwdParams<T>& fwdParams) { in GetEltwiseFwd()
210 MklPrimitive* op) { in SetEltwiseFwd()
240 class MklEltwiseBwdPrimitive : public MklPrimitive {
243 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklEltwiseBwdPrimitive()
416 MklPrimitive* GetEltwiseBwd(const MklEltwiseBwdParams<T>& bwdParams) { in GetEltwiseBwd()
422 MklPrimitive* op) { in SetEltwiseBwd()
|
| D | mkl_conv_grad_filter_ops.cc | 86 class MklConvBwdFilterPrimitive : public MklPrimitive {
90 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklConvBwdFilterPrimitive()
346 MklPrimitive* GetConvBwdFilter( in GetConvBwdFilter()
353 MklPrimitive* op) { in SetConvBwdFilter()
|
| D | mkl_fused_batch_norm_op.cc | 63 class MklFusedBatchNormFwdPrimitive : public MklPrimitive {
66 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklFusedBatchNormFwdPrimitive()
365 MklPrimitive* GetBatchNormFwd(const MklBatchNormFwdParams& fwdParams) { in GetBatchNormFwd()
371 MklPrimitive* op) { in SetBatchNormFwd()
400 class MklFusedBatchNormBwdPrimitive : public MklPrimitive {
403 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklFusedBatchNormBwdPrimitive()
622 MklPrimitive* GetBatchNormBwd(const MklBatchNormBwdParams& bwdParams) { in GetBatchNormBwd()
628 MklPrimitive* op) { in SetBatchNormBwd()
|
| D | mkl_concat_op.cc | 263 class MklConcatFwdPrimitive : public MklPrimitive {
267 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklConcatFwdPrimitive()
427 MklPrimitive* GetConcatFwd(const MklConcatFwdParams& concat_fwd_dims) { in GetConcatFwd()
433 MklPrimitive* op) { in SetConcatFwd()
|
| D | mkl_conv_ops.cc | 98 class MklConvFwdPrimitive : public MklPrimitive {
101 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklConvFwdPrimitive()
401 MklPrimitive* GetConvFwd(const MklConvFwdParams& convFwdDims) { in GetConvFwd()
406 void SetConvFwd(const MklConvFwdParams& convFwdDims, MklPrimitive* op) { in SetConvFwd()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1671 class MklPrimitive {
1673 virtual ~MklPrimitive() {} in ~MklPrimitive()
1674 MklPrimitive() {} in MklPrimitive() function
1675 MklPrimitive(const engine& cpu_engine) { cpu_engine_ = cpu_engine; } in MklPrimitive() function
1796 MklPrimitive* GetOp(const string& key) { in GetOp()
1801 void SetOp(const string& key, MklPrimitive* op) { in SetOp()
1823 static inline LRUCache<MklPrimitive>& GetLRUCache() { in GetLRUCache()
1825 static thread_local LRUCache<MklPrimitive> lru_cache_(kCapacity); in GetLRUCache()
1863 class MklReorderPrimitive : public MklPrimitive {
1866 : MklPrimitive(engine(engine::kind::cpu, 0)) { in MklReorderPrimitive()
[all …]
|