Searched refs:src1_mkl_shape (Results 1 – 1 of 1) sorted by relevance
55 MklDnnShape src1_mkl_shape, src2_mkl_shape; in Compute() local56 GetMklShape(ctx, src1_idx, &src1_mkl_shape); in Compute()58 bool input1_in_mkl_format = src1_mkl_shape.IsMklTensor(); in Compute()60 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()136 md1 = src1_mkl_shape.GetMklLayout(); in Compute()138 memory::format src1_mkl_data_format = src1_mkl_shape.GetTfDataFormat(); in Compute()169 md1 = src1_mkl_shape.GetMklLayout(); in Compute()224 src1_mkl_shape.GetSizesAsMklDnnDims(), in Compute()225 src1_mkl_shape.GetTfDataFormat()); in Compute()