Home
last modified time | relevance | path

Searched refs:IsMklChannelDim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_lrn_op.cc109 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() - in Compute()
361 input_grad_dnn_shape.IsMklChannelDim( in Compute()
363 orig_input_dnn_shape.IsMklChannelDim( in Compute()
365 orig_output_dnn_shape.IsMklChannelDim( in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h253 bool IsMklChannelDim(int d) const { return tf_dim_idx(d) == MklDims::C; } in IsMklChannelDim() function
269 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNCHWFormat()
279 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNHWCFormat()
713 inline bool IsMklChannelDim(int d) const { in IsMklChannelDim() function
737 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNCHWFormat()
747 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNHWCFormat()