Searched defs:transpose_dims (Results 1 – 11 of 11) sorted by relevance
56 std::vector<int64_t> transpose_dims(rank); in Run() local
72 absl::Span<const int64_t> transpose_dims) { in TransposeDims()
199 std::vector<int64_t> transpose_dims; member652 std::vector<int64_t> transpose_dims(activations->shape().rank()); in BringSpaceNextToBatch() local771 std::vector<int64_t> transpose_dims(new_dimensions.size()); in SplitAndTransposeMergedBatch() local2434 std::vector<int64_t> transpose_dims(permute_dims); in BatchToSpace() local2678 std::vector<int64_t> transpose_dims(rank); in PropagateOnConv() local2947 std::vector<int64_t> transpose_dims = retval.transpose_dims; in SplitSpace() local3215 std::vector<int64_t> transpose_dims = retval.transpose_dims; in PropagateOnBackpropFilterConv() local3753 std::vector<int64_t> transpose_dims(rank); in PerformSpaceToBatchOnConvolution() local
2487 const auto& transpose_dims = transpose->dimensions(); in OptimizeDotOfReorderContractingDims() local
531 std::vector<int64_t> transpose_dims(rhs->shape().dimensions_size()); in MatchBackwardInput() local632 std::vector<int64_t> transpose_dims(lhs->shape().dimensions_size()); in ConvertBatchGroupedToFeatureGroupedConvolution() local
294 const auto& transpose_dims = transpose->dimensions(); in TrySimplifyPadding() local
77 std::vector<int64_t> transpose_dims(num_dims + 1); in TransposeFilterForGroupConvolutionBackpropInput() local
452 std::vector<int64_t> transpose_dims = broadcasted_dims; in TridiagonalMatMul() local
516 std::vector<int64_t> transpose_dims(output_rank); in Einsum() local
743 Output transpose_dims = Concat(scope, in GatherV2Grad() local
1847 std::vector<int64_t> transpose_dims( in TryComplexReshardHandling() local