HomeSort by relevance Sort by last modified time
    Searched refs:input_in_mkl_format (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_maxpooling_op.cc 81 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
85 if (input_in_mkl_format == false) {
155 bool input_in_mkl_format = input_shape.IsMklTensor(); local
157 if (input_in_mkl_format == false) {
182 bool input_in_mkl_format = input_shape.IsMklTensor(); local
184 if (!input_in_mkl_format) {
244 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
246 if (input_in_mkl_format == false)
252 if (input_in_mkl_format == false) {
320 bool input_in_mkl_format = input_shape.IsMklTensor() local
393 bool input_in_mkl_format = input_shape.IsMklTensor(); local
472 bool input_in_mkl_format = input_shape.IsMklTensor(); local
    [all...]
mkl_reshape_op.cc 70 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); variable
71 const int64 nelems = input_in_mkl_format
121 if (input_in_mkl_format) {
mkl_avgpooling_op.cc 72 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
74 if (!input_in_mkl_format)
80 if (!input_in_mkl_format) {
102 if (input_in_mkl_format == false) {
163 bool input_in_mkl_format = input_shape.IsMklTensor(); local
165 if (!input_in_mkl_format) {
201 bool input_in_mkl_format = input_shape.IsMklTensor(); local
202 if (!input_in_mkl_format) {
mkl_conv_ops.cc 444 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
459 if (!input_in_mkl_format) {
477 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'C')
489 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'H')
501 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'W')
512 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'N')
559 mkl_context.in_dims = input_in_mkl_format
706 bool input_in_mkl_format = input_shape.IsMklTensor(); local
707 if (input_in_mkl_format) {
826 bool input_in_mkl_format = input_shape.IsMklTensor(); local
    [all...]

Completed in 58 milliseconds