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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_filter_ops.cc 534 auto bwd_pd = convolution_backward_weights::primitive_desc( local
538 AllocateOutputTensor(context, bwd_pd, bwd_output_dims, bwd_output_format,
559 PrepareAndExecutePrimitive(bwd_pd, input, outbackprop, output, bias_grad);
561 PrepareAndExecutePrimitive(bwd_pd, input, outbackprop, output);
mkl_conv_grad_input_ops.cc 439 auto bwd_pd = convolution_backward_data::primitive_desc( local
443 AllocateOutputTensor(context, bwd_pd, bwd_output_dims, bwd_output_format,
449 PrepareAndExecutePrimitive(bwd_pd, filter, outbackprop, output);

Completed in 76 milliseconds