Searched refs:GetMklTensorDim (Results 1 – 2 of 2) sorted by relevance
477 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'C') in Compute()489 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'H') in Compute()501 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'W') in Compute()512 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'N') in Compute()
1132 inline int64 GetMklTensorDim(const MklShape& mkl_shape, char dimension) { in GetMklTensorDim() function