Searched refs:split_vec_shape (Results 1 – 1 of 1) sorted by relevance
810 TensorShape split_vec_shape; in CreateConcatAndSplitNodesForInputTensor() local811 split_vec_shape.AddDim(1); in CreateConcatAndSplitNodesForInputTensor()812 split_vec_shape.set_dim(0, last_dim_vec.size()); in CreateConcatAndSplitNodesForInputTensor()814 Tensor split_vec_tensor(DT_INT32, split_vec_shape); in CreateConcatAndSplitNodesForInputTensor()995 TensorShape split_vec_shape; in CreateConcatAndSplitNodesForOutputTensor() local996 split_vec_shape.AddDim(1); in CreateConcatAndSplitNodesForOutputTensor()997 split_vec_shape.set_dim(0, last_dim_vec.size()); in CreateConcatAndSplitNodesForOutputTensor()999 Tensor split_vec_tensor(DT_INT32, split_vec_shape); in CreateConcatAndSplitNodesForOutputTensor()