Searched refs:kOpDataFormatDimMap (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer.cc | 65 constexpr char kOpDataFormatDimMap[] = "DataFormatDimMap"; variable
525 } else if (op == kOpDataFormatVecPermute || op == kOpDataFormatDimMap) { in UpdateEdge()
1363 UpdateFaninEdgesWithOp(context, {axis_node}, node, kOpDataFormatDimMap)); in TransposeNode()
1554 UpdateFaninEdgesWithOp(context, {1}, node, kOpDataFormatDimMap)); in TransposeNode()
1571 UpdateFaninEdgesWithOp(context, {1}, node, kOpDataFormatDimMap)); in TransposeNode()
1687 UpdateFaninEdgesWithOp(context, {0}, node, kOpDataFormatDimMap)); in TransposeNode()
1703 UpdateFaninEdgesWithOp(context, {2}, node, kOpDataFormatDimMap)); in TransposeNode()
2176 return op == kOpDataFormatDimMap || op == kOpDataFormatVecPermute; in IsDataFormatOp()
|