Searched defs:split_dim_node (Results 1 – 2 of 2) sorted by relevance
751 Node* split_dim_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local937 Node* split_dim_node = nullptr; in CreateConcatAndSplitNodesForOutputTensor() local1451 Node* split_dim_node = init_graph->AddNode(split_dim_def, &status); in InitializeShardedVarOnTPU() local
757 Node* split_dim_node = graph->AddNode(split_dim_def, &s); in CreateSplitNode() local