Searched refs:SanityCheckInputs (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_lrn_op.cc761 SanityCheckInputs(context);
959 void SanityCheckInputs(OpKernelContext* context) {
1010 SanityCheckInputs(context);
1278 void SanityCheckInputs(OpKernelContext* context) {
H A Dmkl_avgpooling_op.cc561 SanityCheckInputs(context, tensor_in_shape, input_gradient_tensor,
670 void SanityCheckInputs(OpKernelContext* context, function in class:tensorflow::MklAvgPoolingGradOp
H A Dmkl_maxpooling_op.cc637 SanityCheckInputs(context, orig_input_tensor, orig_output_tensor,
741 void SanityCheckInputs(OpKernelContext* context, function in class:tensorflow::MklMaxPoolingGradOp

Completed in 133 milliseconds