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

  /external/tensorflow/tensorflow/python/kernel_tests/
depthtospace_op_test.py 39 input_nhwc = math_ops.to_float(inputs)
42 x_tf = array_ops.depth_to_space(input_nhwc, block_size)
47 x_tf = array_ops.depth_to_space(input_nhwc, block_size)
50 input_nchw = test_util.NHWCToNCHW(input_nhwc)
spacetodepth_op_test.py 38 input_nhwc = math_ops.to_float(inputs)
41 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
46 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
49 input_nchw = test_util.NHWCToNCHW(input_nhwc)

Completed in 81 milliseconds