Home
last modified time | relevance | path

Searched refs:split_vec_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc814 Tensor split_vec_tensor(DT_INT32, split_vec_shape); in CreateConcatAndSplitNodesForInputTensor() local
816 split_vec_tensor.flat<int>()(i) = last_dim_vec[i]; in CreateConcatAndSplitNodesForInputTensor()
818 VLOG(3) << "split_vec_tensor: " << split_vec_tensor.DebugString(); in CreateConcatAndSplitNodesForInputTensor()
823 .Attr("value", split_vec_tensor) in CreateConcatAndSplitNodesForInputTensor()
999 Tensor split_vec_tensor(DT_INT32, split_vec_shape); in CreateConcatAndSplitNodesForOutputTensor() local
1001 split_vec_tensor.flat<int>()(i) = last_dim_vec[i]; in CreateConcatAndSplitNodesForOutputTensor()
1003 VLOG(3) << "split_vec_tensor: " << split_vec_tensor.DebugString(); in CreateConcatAndSplitNodesForOutputTensor()
1008 .Attr("value", split_vec_tensor) in CreateConcatAndSplitNodesForOutputTensor()