Home
last modified time | relevance | path

Searched refs:new_window (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc344 Window new_window = conv_window; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
447 new_window.mutable_dimensions(i)->set_padding_low(new_padding_low); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
448 new_window.mutable_dimensions(i)->set_padding_high(new_padding_high); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
449 new_window.mutable_dimensions(i)->set_size( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
468 int64 shard_size_with_halo = new_window.dimensions(i).size(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
484 new_window.dimensions(i).size(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
500 auto conv, create_sharded_conv(conv_lhs, rhs_with_halo, b, new_window)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
653 Window new_window = window; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
697 new_window.mutable_dimensions(i)->set_padding_low(0); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
698 new_window.mutable_dimensions(i)->set_padding_high(0); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
[all …]
Ddot_handler.cc142 auto new_window = original_dot->window(); in GenNewWindow() local
148 auto wd = new_window.mutable_dimensions(i); in GenNewWindow()
170 auto wd = new_window.mutable_dimensions(i); in GenNewWindow()
179 WindowDimension* new_dim = new_window.add_dimensions(); in GenNewWindow()
217 VLOG(2) << "new_window: " << new_window.ShortDebugString(); in GenNewWindow()
218 return new_window; in GenNewWindow()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_4d_expander.cc69 Window new_window; in ExpandInstruction() local
85 *new_window.add_dimensions() = instruction->window().dimensions(i); in ExpandInstruction()
167 instruction->set_window(new_window); in ExpandInstruction()
Dspace_to_batch_converter.cc2009 auto new_window = convolution->window(); in PropagateOnConv() local
2010 new_window.mutable_dimensions(get_chosen_spatial_dim(convolution)) in PropagateOnConv()
2012 new_window.mutable_dimensions(get_chosen_spatial_dim(convolution)) in PropagateOnConv()
2019 new_window, new_dim_numbers, convolution->precision_config(), in PropagateOnConv()
2511 auto new_window = convolution->window(); in PropagateOnBackpropFilterConv() local
2512 new_window.mutable_dimensions(get_chosen_spatial_dim(convolution)) in PropagateOnBackpropFilterConv()
2514 new_window.mutable_dimensions(get_chosen_spatial_dim(convolution)) in PropagateOnBackpropFilterConv()
2516 new_window.mutable_dimensions(get_chosen_spatial_dim(convolution)) in PropagateOnBackpropFilterConv()
2520 auto window_dim = new_window.add_dimensions(); in PropagateOnBackpropFilterConv()
2533 convolution->batch_group_count(), new_window, new_dim_numbers, in PropagateOnBackpropFilterConv()
[all …]
Dconvolution_group_converter.cc530 auto new_window = convolution->window(); in HandleConvolution() local
531 auto new_dim = new_window.add_dimensions(); in HandleConvolution()
570 new_window, dim_numbers, convolution->precision_config())); in HandleConvolution()
Dalgebraic_simplifier.cc4969 Window new_window = window; in HandleReduceWindow() local
4975 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow()
4991 /*window=*/new_window, in HandleReduceWindow()
5174 Window new_window = window; in FoldConvInputPad() local
5176 auto& w = *new_window.mutable_dimensions(dim); in FoldConvInputPad()
5203 new_conv->set_window(new_window); in FoldConvInputPad()
5240 Window new_window = convolution->window(); in FoldConvFilterPad() local
5242 auto& w = *new_window.mutable_dimensions(dim); in FoldConvFilterPad()
5269 new_conv->set_window(new_window); in FoldConvFilterPad()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter.cc404 Window new_window = old_window; in MatchBackwardInput() local
412 auto dim = new_window.mutable_dimensions(i); in MatchBackwardInput()
543 return std::make_tuple(true, new_window, dnums, rhs); in MatchBackwardInput()
605 return std::make_tuple(true, new_window, dnums, rhs); in MatchBackwardInput()