Searched refs:remaining_batch_tensor_idx (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_functor_gpu.cu.cc51 int32 remaining_batch_tensor_idx = batch_tensor_idx; local
57 remaining_batch_tensor_idx % args.batch_tensor_shape[dim];
58 remaining_batch_tensor_idx /= args.batch_tensor_shape[dim];
60 batch_tensor_pos[0] = remaining_batch_tensor_idx;

Completed in 95 milliseconds