Searched refs:src2_dims_size (Results 1 – 1 of 1) sorted by relevance
62 int src2_dims_size = input2_in_mkl_format ? src2_mkl_shape.GetDimension() in Compute() local109 int tmp_size = input1_in_mkl_format ? src2_dims_size : src1_dims_size; in Compute()227 output_mkl_shape.SetTfLayout(src2_dims_size, in Compute()