Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatch_dot_simplification.cc71 DotDimensionNumbers new_dim_numbers = dim_numbers; in ElideDegenerateBatchDimensionFromBatchDot() local
72 new_dim_numbers.clear_lhs_batch_dimensions(); in ElideDegenerateBatchDimensionFromBatchDot()
73 new_dim_numbers.clear_rhs_batch_dimensions(); in ElideDegenerateBatchDimensionFromBatchDot()
78 new_dim_numbers.add_lhs_batch_dimensions(i); in ElideDegenerateBatchDimensionFromBatchDot()
79 new_dim_numbers.add_rhs_batch_dimensions(i); in ElideDegenerateBatchDimensionFromBatchDot()
82 new_dim_numbers.set_lhs_contracting_dimensions( in ElideDegenerateBatchDimensionFromBatchDot()
84 new_dim_numbers.lhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
85 new_dim_numbers.set_rhs_contracting_dimensions( in ElideDegenerateBatchDimensionFromBatchDot()
87 new_dim_numbers.rhs_contracting_dimensions(0) - degenerate_dims.size()); in ElideDegenerateBatchDimensionFromBatchDot()
91 MakeDotHlo(new_lhs, new_rhs, new_dim_numbers, in ElideDegenerateBatchDimensionFromBatchDot()
Dspace_to_batch_converter.cc656 ConvolutionDimensionNumbers new_dim_numbers = dim_numbers; in BringSpaceNextToBatch() local
674 new_dim_numbers.set_kernel_output_feature_dimension(pushed_counter); in BringSpaceNextToBatch()
679 new_dim_numbers.set_kernel_spatial_dimensions(j, pushed_counter); in BringSpaceNextToBatch()
691 new_dim_numbers.set_kernel_input_feature_dimension(activations_batch_dim); in BringSpaceNextToBatch()
706 new_dim_numbers.set_input_batch_dimension(pushed_counter); in BringSpaceNextToBatch()
708 new_dim_numbers.set_input_feature_dimension(pushed_counter); in BringSpaceNextToBatch()
713 new_dim_numbers.set_input_spatial_dimensions(j, pushed_counter); in BringSpaceNextToBatch()
726 new_dim_numbers.set_input_feature_dimension(activations_batch_dim); in BringSpaceNextToBatch()
728 new_dim_numbers.set_input_batch_dimension(activations_batch_dim); in BringSpaceNextToBatch()
732 dim_numbers = new_dim_numbers; in BringSpaceNextToBatch()
[all …]
Dindexed_array_analysis.cc1050 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedLhs() local
1051 new_dim_numbers.set_lhs_contracting_dimensions( in ComputeArrayForDotWithIndexedLhs()
1057 new_dim_numbers, precision_config, lhs->literal(), *rhs->literal()))); in ComputeArrayForDotWithIndexedLhs()
1086 DotDimensionNumbers new_dim_numbers = dim_numbers; in ComputeArrayForDotWithIndexedRhs() local
1087 new_dim_numbers.set_rhs_contracting_dimensions( in ComputeArrayForDotWithIndexedRhs()
1093 new_dim_numbers, precision_config, *lhs->literal(), rhs->literal()))); in ComputeArrayForDotWithIndexedRhs()