Searched refs:GetUsrMem (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_dequantize_op.cc | 151 ReorderPd(cpu_engine, src.GetUsrMem()->get_desc(), cpu_engine, in Compute()
152 dst.GetUsrMem()->get_desc(), attr); in Compute()
155 reorder_net_args.push_back({{MKLDNN_ARG_FROM, *src.GetUsrMem()}, in Compute()
156 {MKLDNN_ARG_TO, *dst.GetUsrMem()}}); in Compute()
|
| D | mkl_transpose_op.cc | 146 auto* prim = FindOrCreateReorder<T>(in.GetUsrMem(), out.GetUsrMem()); in MKLTransposeND()
152 net_args.push_back({{MKLDNN_ARG_FROM, *in.GetUsrMem()}, in MKLTransposeND()
153 {MKLDNN_ARG_TO, *out.GetUsrMem()}}); in MKLTransposeND()
|
| D | mkl_quantize_op.cc | 474 MklReorderWithScalePrimitiveFactory<T>::Get(src.GetUsrMem(), in Compute()
475 dst.GetUsrMem(), fwdParams); in Compute()
|
| D | mkl_slice_op.cc | 450 MklSliceParams sliceParams(&src.GetOpMem(), output.GetUsrMem(), in ComputeMklSlice()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1371 inline const memory* GetUsrMem() const { return user_memory_; } in GetUsrMem() function
|