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

  /external/tensorflow/tensorflow/core/kernels/
mkl_maxpooling_op.cc 695 MklDnnShape orig_input_mkl_shape, grad_mkl_shape; variable
697 GetMklShape(context, kInputTensorIndexGradient, &grad_mkl_shape);
723 grad_mkl_shape.IsMklTensor()
724 ? grad_mkl_shape.GetSizesAsMklDnnDims()
747 grad_mkl_shape.IsMklTensor()
748 ? grad_mkl_shape.GetMklLayout()
    [all...]
mkl_avgpooling_op.cc 581 MklDnnShape orig_input_mkl_shape, grad_mkl_shape; variable
583 GetMklShape(context, kInputTensorIndexInputGradient, &grad_mkl_shape);
612 grad_mkl_shape.IsMklTensor()
613 ? grad_mkl_shape.GetSizesAsMklDnnDims()
636 grad_mkl_shape.IsMklTensor()
637 ? grad_mkl_shape.GetMklLayout()

Completed in 7416 milliseconds