Searched refs:mkl_shape_output (Results 1 – 2 of 2) sorted by relevance
159 MklDnnShape mkl_shape_output; in Compute() local160 mkl_shape_output.SetMklTensor(false); in Compute()163 shape_to, mkl_shape_output); in Compute()
884 MklDnnShape mkl_shape_output; in CopyTfTensorInToOutWithShape() local885 mkl_shape_output.SetMklTensor(false); in CopyTfTensorInToOutWithShape()886 AllocateOutputSetMklShape(context, idx_out, mkl_shape_output); in CopyTfTensorInToOutWithShape()931 MklDnnShape mkl_shape_output; in SetDummyMklDnnShapeOutput() local932 mkl_shape_output.SetMklTensor(false); in SetDummyMklDnnShapeOutput()933 AllocateOutputSetMklShape(context, idx_data_out, mkl_shape_output); in SetDummyMklDnnShapeOutput()