Home
last modified time | relevance | path

Searched defs:input_in_mkl_format (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_maxpooling_op.cc81 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local
155 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives() local
182 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local
244 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local
320 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayouts() local
393 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklPrepareInputs() local
472 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local
Dmkl_avgpooling_op.cc72 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local
163 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives() local
201 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local
Dmkl_reshape_op.cc70 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); in Compute() local
Dmkl_conv_ops.cc444 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local
706 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateInputLayouts() local
826 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local