Searched refs:src2_mkl_shape (Results 1 – 1 of 1) sorted by relevance
55 MklDnnShape src1_mkl_shape, src2_mkl_shape; in Compute() local57 GetMklShape(ctx, src2_idx, &src2_mkl_shape); in Compute()59 bool input2_in_mkl_format = src2_mkl_shape.IsMklTensor(); in Compute()62 int src2_dims_size = input2_in_mkl_format ? src2_mkl_shape.GetDimension() in Compute()68 src2_shape = input2_in_mkl_format ? src2_mkl_shape.GetTfShape() in Compute()152 memory::format src2_mkl_data_format = src2_mkl_shape.GetTfDataFormat(); in Compute()165 md2 = src2_mkl_shape.GetMklLayout(); in Compute()170 md2 = src2_mkl_shape.GetMklLayout(); in Compute()228 src2_mkl_shape.GetSizesAsMklDnnDims(), in Compute()229 src2_mkl_shape.GetTfDataFormat()); in Compute()