Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc1391 auto num_split_attr = op->getAttrOfType<IntegerAttr>("num_split"); in matchAndRewrite() local
1408 auto num_split_attr = op->getAttrOfType<IntegerAttr>("num_split"); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer.cc2133 const auto* num_split_attr = node.GetAttr(kAttrNumSplit); in GetDataFanoutPorts() local