Searched defs:transpose_dims (Results 1 – 6 of 6) sorted by relevance
96 std::vector<int64> transpose_dims(lhs->shape().dimensions_size()); in ConvertBatchGroupedToFeatureGroupedConvolution() local583 std::vector<int64> transpose_dims(rhs->shape().dimensions_size()); in MatchBackwardInput() local
157 std::vector<int64> transpose_dims; member482 std::vector<int64> transpose_dims(activations->shape().rank()); in BringSpaceNextToBatch() local1750 std::vector<int64> transpose_dims(permute_dims); in BatchToSpace() local1984 std::vector<int64> transpose_dims(rank); in PropagateOnConv() local2104 std::vector<int64> transpose_dims = retval.transpose_dims; in SplitSpace() local2352 std::vector<int64> transpose_dims = retval.transpose_dims; in PropagateOnBackpropFilterConv() local2748 std::vector<int64> transpose_dims(rank); in PerformSpaceToBatchOnConvolution() local
54 const auto& transpose_dims = transpose.dimensions(); in CanFoldOperandsIntoDot() local
2179 const auto& transpose_dims = transpose->dimensions(); in OptimizeDotOfReorderContractingDims() local
77 std::vector<int64> transpose_dims(num_dims + 1); in TransposeFilterForGroupConvolutionBackpropInput() local
483 std::vector<int64> transpose_dims(output_rank); in Einsum() local