Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:block_size_sq (Results 1 - 5 of 5) sorted by null

/external/tensorflow/tensorflow/python/kernel_tests/
depthtospace_op_test.py	211 block_size_sq = block_size * block_size 214 assert ic % block_size_sq == 0, (ic, block_size_sq) 215 ow, oh, oc = iw * block_size, ih * block_size, ic // block_size_sq 222 assert ic % block_size_sq == 0, (ic, block_size_sq) 223 ow, oh, oc = iw * block_size, ih * block_size, ic // block_size_sq 310 block_size_sq = block_size * block_size 311 data = np.random.normal(0, 1, b * h * w * d * block_size_sq).astype( 314 x = data.reshape([b, h, w, d * block_size_sq])     [all...]
spacetodepth_op_test.py	206 block_size_sq = block_size * block_size 211 ow, oh, oc = iw // block_size, ih // block_size, ic * block_size_sq 220 ow, oh, oc = iw // block_size, ih // block_size, ic * block_size_sq 303 block_size_sq = block_size * block_size 304 data = np.random.normal(0, 1, b * h * w * d * block_size_sq).astype(
batchtospace_op_test.py	255 block_size_sq = block_size * block_size 257 block_size_sq).astype(np.float32).reshape(
spacetobatch_op_test.py	514 block_size_sq = block_size * block_size 516 block_size_sq).astype(np.float32).reshape(
/external/tensorflow/tensorflow/core/kernels/
depthtospace_op.cc	92 const int block_size_sq = block_size_ * block_size_; variable 96 context, input_depth % block_size_sq == 0, 98 " should be divisible by: ", block_size_sq)); 100 const int output_depth = input_depth / block_size_sq;

Completed in 103 milliseconds