Searched refs:SPLIT_TENSOR (Results 1 – 2 of 2) sorted by relevance
177 constexpr char SPLIT_TENSOR[] = "split_tensor"; variable
625 InsertGetTensorSliceOp(op, next_node, func_graph, index, SPLIT_TENSOR); in SplitTensor()672 std::vector<AnfNodePtr> node_input = CreateInput(op, value_node, SPLIT_TENSOR); in SplitTensorList()678 new_node_prim->set_instance_name(SPLIT_TENSOR); in SplitTensorList()