Searched refs:mkl_shape_output (Results 1 – 2 of 2) sorted by relevance
160 MklDnnShape mkl_shape_output; in Compute() local161 mkl_shape_output.SetMklTensor(false); in Compute()164 shape_to, mkl_shape_output); in Compute()
843 MklDnnShape mkl_shape_output; in CopyTfTensorInToOutWithShape() local844 mkl_shape_output.SetMklTensor(false); in CopyTfTensorInToOutWithShape()845 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()890 MklDnnShape mkl_shape_output; in SetDummyMklDnnShapeOutput() local891 mkl_shape_output.SetMklTensor(false); in SetDummyMklDnnShapeOutput()892 AllocateOutputSetMklShape(context, idx_data_out, mkl_shape_output); in SetDummyMklDnnShapeOutput()