Searched refs:split_vec_tensor (Results 1 – 1 of 1) sorted by relevance
764 Tensor split_vec_tensor(DT_INT32, split_vec_shape); in CreateConcatAndSplitNodesForInputTensor() local766 split_vec_tensor.flat<int>()(i) = last_dim_vec[i]; in CreateConcatAndSplitNodesForInputTensor()768 VLOG(3) << "split_vec_tensor: " << split_vec_tensor.DebugString(); in CreateConcatAndSplitNodesForInputTensor()773 .Attr("value", split_vec_tensor) in CreateConcatAndSplitNodesForInputTensor()949 Tensor split_vec_tensor(DT_INT32, split_vec_shape); in CreateConcatAndSplitNodesForOutputTensor() local951 split_vec_tensor.flat<int>()(i) = last_dim_vec[i]; in CreateConcatAndSplitNodesForOutputTensor()953 VLOG(3) << "split_vec_tensor: " << split_vec_tensor.DebugString(); in CreateConcatAndSplitNodesForOutputTensor()958 .Attr("value", split_vec_tensor) in CreateConcatAndSplitNodesForOutputTensor()