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

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 123 static_cast<dnnLayout_t>(mkl_context.input_shape.GetCurLayout());
519 lt_input = static_cast<dnnLayout_t>(ingrad_shape.GetCurLayout());
527 lt_output = static_cast<dnnLayout_t>(inimage_shape.GetCurLayout());
593 static_cast<dnnLayout_t>(inimage_shape.GetCurLayout()))) {
604 static_cast<dnnLayout_t>(inimage_shape.GetCurLayout()))) {
607 static_cast<dnnLayout_t>(inimage_shape.GetCurLayout()),
796 memory::desc src_md = src_dnn_shape.GetCurLayout();
    [all...]
mkl_input_conversion_op.cc 199 mkl_output_mkl_shape.SetMklLayout(mkl_shape->GetCurLayout());
210 mkl_shape->GetCurLayout(),
mkl_conv_grad_bias_ops.cc 169 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout());
mkl_aggregate_ops.cc 193 lt_input1 = static_cast<dnnLayout_t>(input1_shape.GetCurLayout());
201 lt_input2 = static_cast<dnnLayout_t>(input2_shape.GetCurLayout());
mkl_conv_grad_filter_ops.cc 303 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout());
312 static_cast<dnnLayout_t>(out_backprop_shape.GetCurLayout());
mkl_conv_grad_input_ops.cc 257 lt_filter = (dnnLayout_t)filter_shape.GetCurLayout();
265 lt_outbackprop = (dnnLayout_t)outback_shape.GetCurLayout();
mkl_maxpooling_op.cc 161 lt_user_input = (dnnLayout_t)input_shape.GetCurLayout();
326 lt_input_user = (dnnLayout_t)input_shape.GetCurLayout();
336 lt_outbackprop_user = (dnnLayout_t)output_backprop_shape.GetCurLayout();
    [all...]
mkl_relu_op.cc 177 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout());
238 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout());
245 lt_grad = static_cast<dnnLayout_t>(grad_shape.GetCurLayout());
    [all...]
mkl_avgpooling_op.cc 170 lt_user_input = (dnnLayout_t)input_shape.GetCurLayout();
385 lt_user_outbackprop = (dnnLayout_t)out_backprop_shape.GetCurLayout();
mkl_fused_batch_norm_op.cc 269 static_cast<dnnLayout_t>(mkl_shape_input_shape.GetCurLayout());
551 static_cast<dnnLayout_t>(mkl_shape_input_shape.GetCurLayout());
562 static_cast<dnnLayout_t>(mkl_shape_out_backprop.GetCurLayout());
    [all...]
mkl_concat_op.cc 437 lt_inputs.push_back((dnnLayout_t)is.GetCurLayout());
    [all...]
mkl_conv_ops.cc 355 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout());
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 153 const dnnLayout_t GetCurLayout() const {
551 inline const memory::desc GetCurLayout() const {
    [all...]

Completed in 189 milliseconds