HomeSort by relevance Sort by last modified time
    Searched refs:nchw_input_shape (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
depthtospace_op_test.py 234 nchw_input_shape = [batch_size, in_channels, in_height, in_width]
250 shape = nchw_input_shape if data_format == "NCHW" else nhwc_input_shape
spacetodepth_op_test.py 232 nchw_input_shape = [batch_size, in_channels, in_height, in_width]
248 shape = nchw_input_shape if data_format == "NCHW" else nhwc_input_shape

Completed in 76 milliseconds