Searched refs:input_in_mkl_format (Results 1 – 2 of 2) sorted by relevance
67 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); in Compute() local68 TensorShape input_shape = input_in_mkl_format ? mkl_shape_input.GetTfShape() in Compute()70 const int64 nelems = input_in_mkl_format ? input_shape.num_elements() in Compute()132 if (input_in_mkl_format && !SkipReorder(mkl_shape_input, shape)) { in Compute()
122 bool input_in_mkl_format = mkl_shape.IsMklTensor(); in Compute() local124 if (input_in_mkl_format) { in Compute()