Searched refs:dnn_shape_output (Results 1 – 2 of 2) sorted by relevance
472 MklDnnShape dnn_shape_output; in CallEigenVersion() local473 dnn_shape_output.SetMklTensor(false); in CallEigenVersion()474 dnn_shape_output.SetDimensions(4); in CallEigenVersion()477 tf_shape_output.AddDim(dnn_shape_output.GetSerializeBufferSize()); in CallEigenVersion()482 dnn_shape_output.SerializeMklDnnShape( in CallEigenVersion()
1227 MklDnnShape dnn_shape_output; in ForwardTfTensorInToOut() local1228 dnn_shape_output.SetMklTensor(false); in ForwardTfTensorInToOut()1229 AllocateOutputSetMklShape(context, idx_out, dnn_shape_output); in ForwardTfTensorInToOut()