Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreduce_scatter_utils.cc488 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in MatchReduceScatter() local
H A Dalgebraic_simplifier.cc2447 auto unmodified_dims = ShapeUtil::DimensionsUnmodifiedByReshape( in OptimizeDotOfReorderContractingDims() local
5447 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in HandleReduce() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.cc1230 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in InsertedOrDeleted1SizedDimensions() local
1285 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in ReshapeLeavesDimensionsUnmodified() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc661 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in AddBroadcastSequence() local