Searched refs:middle_size (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_op.cc | 53 const int64 middle_size = input.dim_size(1); in ReverseRows() local
54 const int64 row_size = inner_size * middle_size; in ReverseRows()
65 int remaining = middle_size; in ReverseRows()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 417 for middle_size in list(range(50)) + [100000]:
419 np.arange(outer_size * middle_size * 3, dtype=np.float32),
420 newshape=(outer_size, middle_size, 3))
430 for middle_size in list(range(50)) + [100000]:
432 np.arange(outer_size * middle_size * 4, dtype=np.float32),
433 newshape=(outer_size, middle_size, 4))
442 for middle_size in (1, 2):
444 np.arange(outer_size * middle_size * 3, dtype=np.float32),
445 newshape=(outer_size, middle_size, 3))
|