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

/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_functor_gpu.cu.cc53 int32 batch_tensor_pos[NUM_BLOCK_DIMS + 2]; local
56 batch_tensor_pos[dim] =
60 batch_tensor_pos[0] = remaining_batch_tensor_idx;
62 int32 remaining_block_idx = batch_tensor_pos[0] / args.space_tensor_batch;
63 int32 space_tensor_idx = batch_tensor_pos[NUM_BLOCK_DIMS + 1];
66 batch_tensor_pos[0] % args.space_tensor_batch;
73 batch_tensor_pos[block_dim + 1] * args.block_shape[block_dim] +
H A Dspacetobatch_functor.cc44 for (int64 batch_tensor_pos = 0; batch_tensor_pos < batch_tensor_shape[0];
45 ++batch_tensor_pos) {
47 batch_tensor_pos * block_shape[0] + block_offsets[0] - pad_start[0];

Completed in 99 milliseconds