Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1154 std::vector<std::pair<int64, int64>> unmodified_dims = in InsertedOrDeleted1SizedDimensions() local
1207 std::vector<std::pair<int64, int64>> unmodified_dims = in ReshapeLeavesDimensionsUnmodified() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc2140 auto unmodified_dims = ShapeUtil::DimensionsUnmodifiedByReshape( in OptimizeDotOfReorderContractingDims() local
4647 std::vector<std::pair<int64, int64>> unmodified_dims = in HandleReduce() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc598 std::vector<std::pair<int64, int64>> unmodified_dims = in AddBroadcastSequence() local