Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc608 const xla::OpSharding& sharding, std::map<int, int>* split_dimension_map) { in GetDimensionIndicesAndNumSplitsFromSharding() argument
816 std::map<int, int> split_dimension_map; in CreateOrGetSplitNodesForInputSharding() local
1237 std::map<int, int> split_dimension_map; in CreateConcatNodesForRetval() local