Searched refs:split_tensors (Results 1 – 2 of 2) sorted by relevance
449 std::vector<Tensor> split_tensors; in SplitInputTask() local455 output_task_sizes, &split_tensors); in SplitInputTask()461 if (split_tensors.size() != output_task_sizes.size()) { in SplitInputTask()465 split_tensors.size(), " splits; expected ", output_task_sizes.size()); in SplitInputTask()469 auto moved_tensor_iter = std::next(split_tensors.begin(), j); in SplitInputTask()497 std::map<string, std::vector<Tensor>> split_tensors; in SplitOutputTensors() local
490 split_tensors = []493 split_tensors.append(gather_padded_out_tensor[start_pos:start_pos +495 out_tensor_t = array_ops.concat(split_tensors, 0)