Searched refs:second_shape (Results 1 – 1 of 1) sorted by relevance
966 TensorShape second_shape; in AllocateOutputSetMklShape() local967 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension())); in AllocateOutputSetMklShape()970 second_shape, &second_tensor)); in AllocateOutputSetMklShape()982 TensorShape second_shape; in AllocateOutputSetMklShape() local983 second_shape.AddDim(mkl_shape.GetSerializeBufferSize()); in AllocateOutputSetMklShape()986 second_shape, &second_tensor)); in AllocateOutputSetMklShape()1001 TensorShape second_shape; in AllocateOutputSetMklShape() local1002 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension())); in AllocateOutputSetMklShape()1008 second_shape, &second_tensor)); in AllocateOutputSetMklShape()1022 TensorShape second_shape; in AllocateOutputSetMklShape() local[all …]