Searched refs:CreatePrimitive (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | mkl_conv_ops.h | 487 CreatePrimitive(context, cpu_engine, fwd_pd, &input, &filter, 538 virtual void CreatePrimitive( |
H A D | mkl_conv_grad_input_ops.cc | 418 void CreatePrimitive(OpKernelContext* context, const engine& cpu_engine, function in class:tensorflow::MklConv2DCustomBackpropInputOp |
H A D | mkl_conv_grad_filter_ops.cc | 491 void CreatePrimitive(OpKernelContext* context, const engine& cpu_engine, function in class:tensorflow::MklConv2DCustomBackpropFilterOp |
Completed in 124 milliseconds