Searched refs:src_dnn_data (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_lrn_op.cc | 124 MklDnnData<T> src_dnn_data(&cpu_engine_); in Compute() local
137 src_dnn_data.SetUsrMem(src_md, &src_tensor); in Compute()
138 src_dnn_data.SetOpMemDesc(input_dims, memory::format_tag::nhwc); in Compute()
139 src_dnn_data.SetUsrMemDataHandle(&src_tensor, fwd_stream_); in Compute()
150 src_dnn_data.GetUsrMemDesc(), kernel_size, new_alpha, beta_, bias_); in Compute()
167 src_dnn_data.CheckReorderToOpMem(lrn_prim_desc.src_desc(), cpu_engine_); in Compute()
174 net_args.push_back({{DNNL_ARG_SRC, src_dnn_data.GetOpMem()}, in Compute()
|