Searched refs:space_tensor_batch (Results 1 – 2 of 2) sorted by relevance
34 int32 space_tensor_batch; member62 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 ()()
99 const int64_t space_tensor_batch = space_tensor.dimension(0); in operator ()() local132 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 ()()