Searched refs:input_in_mkl_format (Results 1 – 4 of 4) sorted by relevance
81 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local85 if (input_in_mkl_format == false) { in Compute()155 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives() local157 if (input_in_mkl_format == false) { in MklCreateLayoutsAndPrimitives()182 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local184 if (!input_in_mkl_format) { in MklCleanup()244 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local246 if (input_in_mkl_format == false) in Compute()252 if (input_in_mkl_format == false) { in Compute()320 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayouts() local[all …]
72 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local74 if (!input_in_mkl_format) in Compute()80 if (!input_in_mkl_format) { in Compute()102 if (input_in_mkl_format == false) { in Compute()163 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateLayoutsAndPrimitives() local165 if (!input_in_mkl_format) { in MklCreateLayoutsAndPrimitives()201 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local202 if (!input_in_mkl_format) { in MklCleanup()
70 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); in Compute() local71 const int64 nelems = input_in_mkl_format in Compute()121 if (input_in_mkl_format) { in Compute()
444 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute() local459 if (!input_in_mkl_format) { in Compute()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()559 mkl_context.in_dims = input_in_mkl_format in Compute()706 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCreateInputLayouts() local707 if (input_in_mkl_format) { in MklCreateInputLayouts()826 bool input_in_mkl_format = input_shape.IsMklTensor(); in MklCleanup() local[all …]