Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc135 const auto& transpose_dimensions = transpose.dimensions(); in FoldTransposeIntoConvolution() local
158 const auto& transpose_dimensions = transpose.dimensions(); in FoldTransposeIntoConvolution() local
Dalgebraic_simplifier.cc299 std::vector<int64> transpose_dimensions(batch_dimensions.begin(), in NormalizeDotOperandToBatchMajorAndContractingMinor() local
2974 auto transpose_dimensions = arg->dimensions(); in HandleReduce() local