Searched defs:shape_replacement_map (Results 1 – 1 of 1) sorted by relevance
90 shape_replacement_map) { in MaybeReplaceShapeOrShapeNOp()131 shape_replacement_map) { in MaybeReplaceRankOp()147 shape_replacement_map) { in MaybeReplaceSizeOp()184 shape_replacement_map) { in MaybeReplaceShapeOp()225 shape_replacement_map) { in IsConstantFoldable()297 std::unordered_map<const Node*, std::vector<Tensor>>* shape_replacement_map, in ConsiderConstantFoldableNode()356 shape_replacement_map) { in FindConstantFoldableNodes()418 shape_replacement_map, in AddShapeNodeToConstantGraph()445 shape_replacement_map, in GetConstantGraph()609 std::unordered_map<const Node*, std::vector<Tensor>> shape_replacement_map; in ConstantFold() local