Searched defs:split_tensor (Results 1 – 5 of 5) sorted by relevance
132 void split_tensor(ComputeGraph& graph, const std::vector<ValueRef>& args) { in split_tensor() function
537 def split_tensor(x): function
58 const Tensor& split_tensor = context->input(1); in ComputeEasyCases() local
626 std::vector<Tensor> split_tensor; in SplitOutputTensors() local
737 def split_tensor(func, *args, **kwargs): function