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

  /external/tensorflow/tensorflow/python/ops/
transpose_benchmark.py 122 large_shapes = [[2, 40, 40, 40, 32], [2, 40, 40, 40, 64]] * 2 + [[
136 for ishape, perm in zip(large_shapes, large_perms):
  /external/tensorflow/tensorflow/python/kernel_tests/
transpose_op_test.py 140 large_shapes = [[4, 10, 10, 10, 3], [4, 10, 10, 10, 8], [4, 10, 10, 10, 13],
149 for input_shape, perm in zip(large_shapes, perms):
164 large_shapes = [[4, 10, 10, 3], [4, 10, 10, 8], [4, 10, 10, 13],
170 for input_shape, perm in zip(large_shapes, perms):
222 large_shapes = [[4, 1000, 3], [4, 1000, 8], [4, 1000, 13], [4, 3, 1000],
227 for input_shape, perm in zip(large_shapes, perms):
243 large_shapes = [[1000000, 31, 3], [3, 1000000, 31], [3, 31, 1000000],
248 for input_shape, perm in zip(large_shapes, perms):

Completed in 110 milliseconds