Home
last modified time | relevance | path

Searched refs:space_tensor_batch (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_functor_gpu.cu.cc34 int32 space_tensor_batch; member
62 int32 remaining_block_idx = batch_tensor_pos[0] / args.space_tensor_batch; in S2B()
66 batch_tensor_pos[0] % args.space_tensor_batch; in S2B()
116 args.space_tensor_batch = space_tensor.dimension(0); in operator ()()
Dspacetobatch_functor.cc99 const int64_t space_tensor_batch = space_tensor.dimension(0); in operator ()() local
132 const int64_t space_tensor_b = batch_tensor_b % space_tensor_batch; in operator ()()
133 int64_t block_index = batch_tensor_b / space_tensor_batch; in operator ()()