HomeSort by relevance Sort by last modified time
    Searched refs:original_input_md (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_maxpooling_op.cc 650 memory::desc original_input_md = ConfigureOriginalInput( variable
660 output_dnn_data.SetUsrMem(original_input_md);
665 prop_kind::forward, algorithm::pooling_max, original_input_md,
    [all...]
mkl_avgpooling_op.cc 577 memory::desc original_input_md = ConfigureOriginalInput( variable
603 original_input_md, original_output_md,

Completed in 217 milliseconds