Home
last modified time | relevance | path

Searched refs:dnn_data_input (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_reshape_op.cc140 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute() local
154 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor); in Compute()
170 if (dnn_data_input.CheckReorderToOpMem(output_tf_md, output_tensor, in Compute()
Dmkl_pooling_ops_common.h625 MklDnnData<T>* dnn_data_input) { in ConfigureInput() argument
627 DCHECK(dnn_data_input); in ConfigureInput()
640 dnn_data_input->SetUsrMem(input_md, &input_tensor); in ConfigureInput()
650 dnn_data_input->SetOpMemDesc(input_sizes, this->data_format_mkldnn_); in ConfigureInput()
Dmkl_avgpooling_op.cc58 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute() local
Dmkl_maxpooling_op.cc59 MklDnnData<T> dnn_data_input(&cpu_engine_); in Compute() local