Searched refs:op_pd (Results 1 – 2 of 2) sorted by relevance
47 #define GET_SRC_DESC_FROM_OP_PD(op_pd) op_pd->src_desc() argument48 #define GET_DST_DESC_FROM_OP_PD(op_pd) op_pd->dst_desc() argument49 #define GET_BIAS_DESC_FROM_OP_PD(op_pd) op_pd->bias_desc() argument50 #define GET_DIFF_DST_DESC_FROM_OP_PD(op_pd) op_pd->diff_dst_desc() argument51 #define GET_WORKSPACE_DESC_FROM_OP_PD(op_pd) op_pd->workspace_desc() argument55 #define GET_WEIGHTS_DESC_FROM_OP_PD(op_pd) op_pd->weights_desc() argument56 #define GET_WEIGHTS_FORMAT_FROM_OP_PD(op_pd, op) \ argument57 GET_WEIGHTS_DESC_FROM_OP_PD(op_pd)58 #define IS_DIFF_DST_REORDER_NEEDED(diff_dst_md, op_pd, op) \ argument59 diff_dst_md != op_pd->diff_dst_desc()[all …]
1440 inline bool IsReorderNeeded(const memory::desc& op_pd) const { in IsReorderNeeded() argument1442 return op_pd != user_memory_->get_desc(); in IsReorderNeeded()1599 inline bool CheckReorderToOpMem(const memory::desc& op_pd,1603 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor),1620 inline bool PrepareReorderToUserMemIfReq(const memory::desc& op_pd) { in PrepareReorderToUserMemIfReq() argument1622 if (IsReorderNeeded(op_pd)) { in PrepareReorderToUserMemIfReq()1624 reorder_memory_ = new memory(op_pd, *cpu_engine_); in PrepareReorderToUserMemIfReq()