Searched defs:transpose_dims (Results 1 – 7 of 7) sorted by relevance
96 std::vector<int64> transpose_dims(lhs->shape().dimensions_size()); in ConvertBatchGroupedToFeatureGroupedConvolution() local610 std::vector<int64> transpose_dims(rhs->shape().dimensions_size()); in MatchBackwardInput() local
54 const auto& transpose_dims = transpose.dimensions(); in CanFoldOperandsIntoDot() local
175 std::vector<int64> transpose_dims; member548 std::vector<int64> transpose_dims(activations->shape().rank()); in BringSpaceNextToBatch() local2103 std::vector<int64> transpose_dims(permute_dims); in BatchToSpace() local2348 std::vector<int64> transpose_dims(rank); in PropagateOnConv() local2489 std::vector<int64> transpose_dims = retval.transpose_dims; in SplitSpace() local2734 std::vector<int64> transpose_dims = retval.transpose_dims; in PropagateOnBackpropFilterConv() local3231 std::vector<int64> transpose_dims(rank); in PerformSpaceToBatchOnConvolution() local
2241 const auto& transpose_dims = transpose->dimensions(); in OptimizeDotOfReorderContractingDims() local
77 std::vector<int64> transpose_dims(num_dims + 1); in TransposeFilterForGroupConvolutionBackpropInput() local
516 std::vector<int64> transpose_dims(output_rank); in Einsum() local
742 Output transpose_dims = Concat(scope, in GatherV2Grad() local