Searched refs:op_md (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1470 inline bool CheckReorderToOpMem(const memory::desc& op_md, in CheckReorderToOpMem() argument
1476 if (IsReorderNeeded(op_md)) { in CheckReorderToOpMem()
1478 reorder_memory_ = new memory(op_md, engine); in CheckReorderToOpMem()
1487 inline bool CheckReorderToOpMem(const memory::desc& op_md,
1491 if (IsReorderNeeded(op_md)) {
1495 reorder_memory_ = new memory(op_md, engine);
1524 inline bool CheckReorderToOpMem(const memory::desc& op_md, in CheckReorderToOpMem() argument
1531 if (IsReorderNeeded(op_md)) { in CheckReorderToOpMem()
1533 reorder_memory_ = new memory(op_md, engine, reorder_data_handle); in CheckReorderToOpMem()
1546 inline bool CheckReorderToOpMem(const memory::desc& op_md,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 434 auto op_md = in ComputeMklSlice() local
436 src.CheckReorderToOpMem(op_md, cpu_engine, context); in ComputeMklSlice()
|