Searched defs:orig_output_mkl_shape (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | mkl_maxpooling_op.cc | 630 MklDnnShape orig_input_mkl_shape, orig_output_mkl_shape, grad_mkl_shape, variable
633 GetMklShape(context, kInputTensorIndexOrigOutput, &orig_output_mkl_shape);
639 orig_output_mkl_shape, grad_mkl_shape,
655 pool_params, orig_output_mkl_shape, output_dims_mkl_order);
747 const MklDnnShape& orig_output_mkl_shape,
759 if (!orig_output_mkl_shape.IsMklTensor()) {
764 OP_REQUIRES(context, orig_output_mkl_shape.GetDimension() == 4,
741 SanityCheckInputs(OpKernelContext* context, const Tensor& orig_input_tensor, const Tensor& orig_output_tensor, const Tensor& grad_tensor, const Tensor& workspace_tensor, const MklDnnShape& orig_input_mkl_shape, const MklDnnShape& orig_output_mkl_shape, const MklDnnShape& grad_mkl_shape, const MklDnnShape& workspace_mkl_shape) argument
|
Completed in 84 milliseconds