Home
last modified time | relevance | path

Searched refs:transpose_dims (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc96 std::vector<int64> transpose_dims(lhs->shape().dimensions_size()); in ConvertBatchGroupedToFeatureGroupedConvolution() local
97 std::iota(transpose_dims.begin(), transpose_dims.end(), 0); in ConvertBatchGroupedToFeatureGroupedConvolution()
98 transpose_dims.erase(transpose_dims.begin() + input_batch_dimension); in ConvertBatchGroupedToFeatureGroupedConvolution()
99 transpose_dims.insert(transpose_dims.begin() + input_feature_dimension, in ConvertBatchGroupedToFeatureGroupedConvolution()
102 ComposePermutations(lhs->shape().dimensions(), transpose_dims); in ConvertBatchGroupedToFeatureGroupedConvolution()
105 lhs, transpose_dims)); in ConvertBatchGroupedToFeatureGroupedConvolution()
583 std::vector<int64> transpose_dims(rhs->shape().dimensions_size()); in MatchBackwardInput() local
584 std::iota(transpose_dims.begin(), transpose_dims.end(), 0); in MatchBackwardInput()
585 transpose_dims.erase(transpose_dims.begin() + input_feature_dimension); in MatchBackwardInput()
586 transpose_dims.insert(transpose_dims.begin() + output_feature_dimension, in MatchBackwardInput()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc54 const auto& transpose_dims = transpose.dimensions(); in CanFoldOperandsIntoDot() local
55 if (transpose_dims.size() != expected_rank) { in CanFoldOperandsIntoDot()
64 transpose_dims[i]); in CanFoldOperandsIntoDot()
65 if ((transpose_dims[i] == i) != is_batch) { in CanFoldOperandsIntoDot()
Dspace_to_batch_converter.cc157 std::vector<int64> transpose_dims; member
482 std::vector<int64> transpose_dims(activations->shape().rank()); in BringSpaceNextToBatch() local
484 absl::c_iota(transpose_dims, 0); in BringSpaceNextToBatch()
497 transpose_dims[dim_counter++] = activations_batch_dim; in BringSpaceNextToBatch()
512 transpose_dims[dim_counter++] = i; in BringSpaceNextToBatch()
519 MakeTransposeHlo(activations, transpose_dims)); in BringSpaceNextToBatch()
529 transpose_dims[dim_counter++] = activations_batch_dim; in BringSpaceNextToBatch()
546 transpose_dims[dim_counter++] = i; in BringSpaceNextToBatch()
553 MakeTransposeHlo(activations, transpose_dims)); in BringSpaceNextToBatch()
565 return SpaceNextToBatchDetails{activations, transpose_dims}; in BringSpaceNextToBatch()
[all …]
Dalgebraic_simplifier.cc2179 const auto& transpose_dims = transpose->dimensions(); in OptimizeDotOfReorderContractingDims() local
2180 for (int64 i = 0; i < transpose_dims.size(); ++i) { in OptimizeDotOfReorderContractingDims()
2181 if (transpose_dims[i] != i && in OptimizeDotOfReorderContractingDims()
2190 permutation.push_back(transpose_dims[dim] - lhs_contracting_dims[0]); in OptimizeDotOfReorderContractingDims()
2275 auto inverse_lhs_transpose_dims = InversePermutation(transpose_dims); in OptimizeDotOfReorderContractingDims()
/external/tensorflow/tensorflow/compiler/tests/
Dtest_utils.py43 transpose_dims = [dim_map[d] for d in data_format_dst]
44 return np.transpose(x, transpose_dims)
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc77 std::vector<int64> transpose_dims(num_dims + 1); in TransposeFilterForGroupConvolutionBackpropInput() local
78 std::iota(transpose_dims.begin(), transpose_dims.end(), 0); in TransposeFilterForGroupConvolutionBackpropInput()
79 std::swap(transpose_dims[num_spatial_dims], in TransposeFilterForGroupConvolutionBackpropInput()
80 transpose_dims[num_spatial_dims + 1]); in TransposeFilterForGroupConvolutionBackpropInput()
81 result = xla::Transpose(result, transpose_dims); in TransposeFilterForGroupConvolutionBackpropInput()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc483 std::vector<int64> transpose_dims(output_rank); in Einsum() local
485 transpose_dims[output_transpose_dims[i]] = i; in Einsum()
515 dot = Transpose(dot, transpose_dims); in Einsum()
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py689 transpose_dims = array_ops.concat([
693 values_transpose = array_ops.transpose(values, transpose_dims)