Home
last modified time | relevance | path

Searched defs:op_pd (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_types.h47 #define GET_SRC_DESC_FROM_OP_PD(op_pd) op_pd->src_desc() argument
48 #define GET_DST_DESC_FROM_OP_PD(op_pd) op_pd->dst_desc() argument
49 #define GET_BIAS_DESC_FROM_OP_PD(op_pd) op_pd->bias_desc() argument
50 #define GET_DIFF_DST_DESC_FROM_OP_PD(op_pd) op_pd->diff_dst_desc() argument
51 #define GET_WORKSPACE_DESC_FROM_OP_PD(op_pd) op_pd->workspace_desc() argument
55 #define GET_WEIGHTS_DESC_FROM_OP_PD(op_pd) op_pd->weights_desc() argument
56 #define GET_WEIGHTS_FORMAT_FROM_OP_PD(op_pd, op) \ argument
58 #define IS_DIFF_DST_REORDER_NEEDED(diff_dst_md, op_pd, op) \ argument
60 #define IS_DIFF_FILTER_REORDER_NEEDED(diff_filter_md, fmt, op_pd, op) \ argument
62 #define IS_FILTER_REORDER_NEEDED(filter_md, op_pd, op) \ argument
[all …]
Dmkl_util.h1440 inline bool IsReorderNeeded(const memory::desc& op_pd) const { in IsReorderNeeded()
1620 inline bool PrepareReorderToUserMemIfReq(const memory::desc& op_pd) { in PrepareReorderToUserMemIfReq()