Home
last modified time | relevance | path

Searched defs:split_dim_node (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc751 Node* split_dim_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local
937 Node* split_dim_node = nullptr; in CreateConcatAndSplitNodesForOutputTensor() local
1451 Node* split_dim_node = init_graph->AddNode(split_dim_def, &status); in InitializeShardedVarOnTPU() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc757 Node* split_dim_node = graph->AddNode(split_dim_def, &s); in CreateSplitNode() local