Searched refs:input1_in_mkl_format (Results 1 – 1 of 1) sorted by relevance
58 bool input1_in_mkl_format = src1_mkl_shape.IsMklTensor(); in Compute() local60 int src1_dims_size = input1_in_mkl_format ? src1_mkl_shape.GetDimension() in Compute()66 src1_shape = input1_in_mkl_format ? src1_mkl_shape.GetTfShape() in Compute()80 if (!input1_in_mkl_format && src1_dims_size == 0) { in Compute()93 if (!input1_in_mkl_format && !input2_in_mkl_format) { in Compute()109 int tmp_size = input1_in_mkl_format ? src2_dims_size : src1_dims_size; in Compute()124 if (!input1_in_mkl_format && !input2_in_mkl_format) { in Compute()131 } else if (input1_in_mkl_format && !input2_in_mkl_format) { in Compute()150 } else if (input2_in_mkl_format && !input1_in_mkl_format) { in Compute()182 if (input1_in_mkl_format || input2_in_mkl_format) { in Compute()[all …]