Searched defs:nchw_shape (Results 1 – 3 of 3) sorted by relevance
365 const TensorShape nchw_shape = ShapeFromFormat( in launch() local
1457 TensorShape nchw_shape = {dims.batch_size, dims.out_depth, in Compute() local1850 TensorShape nchw_shape = {dims.batch_size, dims.out_depth, in Compute() local1869 TensorShape nchw_shape = { in Compute() local
991 TensorShape nchw_shape = in operator ()() local