Searched refs:GetConcatAxis (Results 1 – 2 of 2) sorted by relevance
321 bool GetConcatAxis(const NodeDef& node, int* axis);
3762 bool ConstantFolding::GetConcatAxis(const NodeDef& node, int* axis) { in GetConcatAxis() function in tensorflow::grappler::ConstantFolding3788 if (!use_shape_info || !GetConcatAxis(*node, &axis) || in MergeConcat()3808 if (!GetConcatAxis(*parent, &parent_axis) || axis != parent_axis) { in MergeConcat()