Home
last modified time | relevance | path

Searched refs:TransformNHWCToNCHW (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc398 Status TransformNHWCToNCHW(OpKernelContext* ctx, const Tensor& nhwc_tensor, in TransformNHWCToNCHW() function
551 OP_REQUIRES_OK(ctx, (TransformNHWCToNCHW<T, rank>( in launch()
557 ctx, (TransformNHWCToNCHW<T, rank>( in launch()