Searched defs:split_tensors (Results 1 – 2 of 2) sorted by relevance
383 def split_tensors(rt): function402 def split_tensors(rt, split_lengths): function470 def split_tensors(rt, split_lengths): function
556 std::vector<Tensor> split_tensors; in SplitInputTask() local604 std::map<string, std::vector<Tensor>> split_tensors; in SplitOutputTensors() local