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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_filter_ops.cc 525 convolution_direct, input->GetOpMemDesc(),
526 output->GetOpMemDesc(), bias_grad->GetOpMemDesc(),
527 outbackprop->GetOpMemDesc(), strides, padding_l, padding_r,
530 convolution_direct, input->GetOpMemDesc(),
531 output->GetOpMemDesc(), outbackprop->GetOpMemDesc(), strides,
mkl_softmax_op.cc 95 src.GetOpMemDesc(), axis);
mkl_conv_ops.cc 608 prop_kind::forward, convolution_direct, src.GetOpMemDesc(),
609 filter.GetOpMemDesc(), bias.GetOpMemDesc(), output.GetOpMemDesc(),
629 prop_kind::forward, convolution_direct, src.GetOpMemDesc(),
630 filter.GetOpMemDesc(), output.GetOpMemDesc(), strides, padding_l,
    [all...]
mkl_conv_grad_input_ops.cc 436 convolution_direct, output->GetOpMemDesc(), filter->GetOpMemDesc(),
437 outbackprop->GetOpMemDesc(), strides, padding_l, padding_r, padding);
  /external/tensorflow/tensorflow/core/util/
mkl_util.h     [all...]

Completed in 428 milliseconds