Searched refs:nhwc_tensor (Results 1 – 1 of 1) sorted by relevance
398 Status TransformNHWCToNCHW(OpKernelContext* ctx, const Tensor& nhwc_tensor, in TransformNHWCToNCHW() argument407 ctx->eigen_device<GPUDevice>(), nhwc_tensor.tensor<T, NDIMS>(), in TransformNHWCToNCHW()411 CHECK(transformed_tensor->CopyFrom(nhwc_tensor, nchw_shape)); in TransformNHWCToNCHW()