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

  /external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc 90 src.SetOpMemDesc(src_dims, memory::format::nc);
mkl_conv_ops.h 481 input.SetOpMemDesc(fwd_input_dims, memory::format::any);
482 filter.SetOpMemDesc(fwd_filter_dims, memory::format::any);
483 outbackprop.SetOpMemDesc(outbprop_dims, memory::format::any);
484 output.SetOpMemDesc(bwd_output_dims, memory::format::any);
mkl_conv_ops.cc 593 src.SetOpMemDesc(src_dims, memory::format::any);
594 filter.SetOpMemDesc(filter_dims, memory::format::any);
595 output.SetOpMemDesc(output_dims_mkl_order, memory::format::any);
604 bias.SetOpMemDesc(bias_size, memory::format::any);
    [all...]
mkl_lrn_op.cc 804 src_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc);
808 dst_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc);
    [all...]
mkl_conv_grad_filter_ops.cc 518 bias_grad->SetOpMemDesc(bias_grad_dims, memory::format::x);
mkl_fused_batch_norm_op.cc     [all...]
mkl_avgpooling_op.cc 504 dnn_data_output.SetOpMemDesc(output_dims_mkl_order, memory::format::any);
mkl_maxpooling_op.cc 549 dnn_data_output.SetOpMemDesc(output_dims_mkl_order, memory::format::any);
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h     [all...]

Completed in 520 milliseconds