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

  /external/tensorflow/tensorflow/core/kernels/
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_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...]

Completed in 148 milliseconds