Home
last modified time | relevance | path

Searched refs:new_dim_numbers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatch_dot_simplification.cc72 DotDimensionNumbers new_dim_numbers = dim_numbers; in ElideDegenerateBatchDimensionFromBatchDot() local
73 new_dim_numbers.clear_lhs_batch_dimensions(); in ElideDegenerateBatchDimensionFromBatchDot()
74 new_dim_numbers.clear_rhs_batch_dimensions(); in ElideDegenerateBatchDimensionFromBatchDot()
79 new_dim_numbers.add_lhs_batch_dimensions(i); in ElideDegenerateBatchDimensionFromBatchDot()
80 new_dim_numbers.add_rhs_batch_dimensions(i); in ElideDegenerateBatchDimensionFromBatchDot()
83 new_dim_numbers.set_lhs_contracting_dimensions( in ElideDegenerateBatchDimensionFromBatchDot()
85 new_dim_numbers.lhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
86 new_dim_numbers.set_rhs_contracting_dimensions( in ElideDegenerateBatchDimensionFromBatchDot()
88 new_dim_numbers.rhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
92 MakeDotHlo(new_lhs, new_rhs, new_dim_numbers, in ElideDegenerateBatchDimensionFromBatchDot()
Dtranspose_folding.cc112 DotDimensionNumbers new_dim_numbers = dot->dot_dimension_numbers(); in FoldTransposeIntoDot() local
116 CHECK_EQ(new_dim_numbers.lhs_contracting_dimensions_size(), 1); in FoldTransposeIntoDot()
117 CHECK_EQ(new_dim_numbers.rhs_contracting_dimensions_size(), 1); in FoldTransposeIntoDot()
125 new_dim_numbers.set_lhs_contracting_dimensions( in FoldTransposeIntoDot()
127 new_lhs->dimensions(new_dim_numbers.lhs_contracting_dimensions(0))); in FoldTransposeIntoDot()
132 new_dim_numbers.set_rhs_contracting_dimensions( in FoldTransposeIntoDot()
134 new_rhs->dimensions(new_dim_numbers.rhs_contracting_dimensions(0))); in FoldTransposeIntoDot()
140 dot->shape(), new_lhs, new_rhs, new_dim_numbers, dot->precision_config()); in FoldTransposeIntoDot()
Dspace_to_batch_converter.cc486 ConvolutionDimensionNumbers new_dim_numbers = dim_numbers; in BringSpaceNextToBatch() local
504 new_dim_numbers.set_kernel_output_feature_dimension(pushed_counter); in BringSpaceNextToBatch()
509 new_dim_numbers.set_kernel_spatial_dimensions(j, pushed_counter); in BringSpaceNextToBatch()
521 new_dim_numbers.set_kernel_input_feature_dimension(activations_batch_dim); in BringSpaceNextToBatch()
536 new_dim_numbers.set_input_batch_dimension(pushed_counter); in BringSpaceNextToBatch()
538 new_dim_numbers.set_input_feature_dimension(pushed_counter); in BringSpaceNextToBatch()
543 new_dim_numbers.set_input_spatial_dimensions(j, pushed_counter); in BringSpaceNextToBatch()
556 new_dim_numbers.set_input_feature_dimension(activations_batch_dim); in BringSpaceNextToBatch()
558 new_dim_numbers.set_input_batch_dimension(activations_batch_dim); in BringSpaceNextToBatch()
562 dim_numbers = new_dim_numbers; in BringSpaceNextToBatch()
[all …]
Dindexed_array_analysis.cc1046 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedLhs() local
1047 new_dim_numbers.set_lhs_contracting_dimensions( in ComputeArrayForDotWithIndexedLhs()
1053 new_dim_numbers, precision_config, lhs->literal(), *rhs->literal()))); in ComputeArrayForDotWithIndexedLhs()
1082 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedRhs() local
1083 new_dim_numbers.set_rhs_contracting_dimensions( in ComputeArrayForDotWithIndexedRhs()
1089 new_dim_numbers, precision_config, *lhs->literal(), rhs->literal()))); in ComputeArrayForDotWithIndexedRhs()