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

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.cc 520 MklDnnShape src_mkl_shape, filter_mkl_shape; variable
522 GetMklShape(context, kInputIndex_Filter, &filter_mkl_shape);
523 OP_REQUIRES(context, filter_mkl_shape.IsMklTensor() == false,
558 filter_mkl_shape.SetMklTensor(false);
562 filter_mkl_shape);
580 auto filter_md = filter_mkl_shape.IsMklTensor() // Should NEVER be true
581 ? filter_mkl_shape.GetMklLayout()
698 MklDnnShape filter_mkl_shape; local
699 filter_mkl_shape.SetMklTensor(true);
700 filter_mkl_shape.SetMklLayout(&filter_pd)
    [all...]
mkl_conv_ops.h 368 MklDnnShape input_mkl_shape, filter_mkl_shape, outbprop_mkl_shape; variable
370 GetMklShape(context, kFilterIdx, &filter_mkl_shape);
373 ValidateMklShapes(input_mkl_shape, filter_mkl_shape, outbprop_mkl_shape);
434 auto fwd_filter_md = filter_mkl_shape.IsMklTensor()
435 ? filter_mkl_shape.GetMklLayout()
504 const MklDnnShape& filter_mkl_shape,
mkl_conv_grad_filter_ops.cc 448 const MklDnnShape& filter_mkl_shape,
450 CHECK(!filter_mkl_shape.IsMklTensor())
mkl_conv_grad_input_ops.cc 373 const MklDnnShape& filter_mkl_shape,

Completed in 81 milliseconds