| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_window_utils.cc | 94 int64 window_dilation) { in EffectiveFilterSize() argument
95 return ConstantR0<int32>(comp, (window_size - 1) * window_dilation + 1, in EffectiveFilterSize()
102 int64 window_dilation, in GetWindowedOutputSize() argument
110 EffectiveFilterSize(comp, window_size, window_dilation); in GetWindowedOutputSize()
131 int64 window_dilation, in GetWindowedInputGradSize() argument
137 ConstantR0<int32>(comp, (window_size - 1) * window_dilation + 1); in GetWindowedInputGradSize()
140 input_size, window_size, window_dilation, window_stride, padding_type); in GetWindowedInputGradSize()
|
| D | dynamic_window_utils.h | 39 int64 window_dilation,
45 int64 window_dilation,
|
| D | dot_as_convolution_util.cc | 33 wd.window_dilation() == 1) || in ConvSpatialDimensionIsParallel()
34 (std::max<int64>(1, lhs_size - 1) == wd.window_dilation() && in ConvSpatialDimensionIsParallel()
44 wd.window_dilation() == 1 && wd.stride() == lhs_size && in ConvSpatialDimensionIsParallel()
77 wd.window_dilation() == 1 && wd.padding_high() == 0 && in ParseConvolutionDimsInfo()
83 } else if (wd.stride() == 1 && wd.window_dilation() == 1 && in ParseConvolutionDimsInfo()
|
| D | dynamic_padder.cc | 769 window_dim->size(), window_dim->window_dilation()); in RewriteInputWithDynamicPadding()
839 /*window_dilation=*/window_dim.window_dilation(), in RewriteDynamicConvolutionInputGrad()
901 operand_dynamic_size, window_dim.size(), window_dim.window_dilation(), in RewriteDynamicConvolutionForward()
982 /*window_stride=*/window_dim.window_dilation(), in RewriteDynamicConvolutionKernelGrad()
1045 operand_dynamic_size, window_dim.size(), window_dim.window_dilation(), in RewriteDynamicReduceWindowSamePadding()
1100 operand_dynamic_size, window_dim.size(), window_dim.window_dilation(), in RewriteDynamicSelectAndScatterSamePadding()
1128 window_dim.size(), window_dim.window_dilation()); in RewriteDynamicSelectAndScatterSamePadding()
|
| D | hlo_cost_analysis.cc | 631 window_dim.window_dilation() == 1 && in HandleConvolution()
641 window_dim.window_dilation() == 1 && window_dim.base_dilation() == 1 && in HandleConvolution()
660 kernel_idx * window_dim.window_dilation(); in HandleConvolution()
|
| D | dynamic_dimension_inference.cc | 715 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleDynamicConvolutionForward()
734 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleDynamicWindowSamePadding()
1069 dynamic_size, window_dim.size(), window_dim.window_dilation(), in HandleReduceWindow()
|
| D | space_to_batch_converter.cc | 294 .window_dilation() != 1) { in IsConvSuitableForSpaceToBatch()
805 const int64 rhs_dilation = win_dims.window_dilation(); in CanPropagate()
1458 consumer->window().dimensions(dim).window_dilation()); in Propagate()
2153 .window_dilation(); in PropagateOnBackpropFilterConv()
|
| D | algebraic_simplifier.cc | 5258 if (w.window_dilation() > 1) { in FoldConvFilterPad()
5261 CHECK_EQ(w.window_dilation(), 1); in FoldConvFilterPad()
5301 window_dims[spatial_dim].window_dilation() == 1; in SwapConvOperands()
5318 1 + (kernel_size - 1) * window_dims[spatial_dim].window_dilation(); in SwapConvOperands()
5338 new_dim->set_base_dilation(window_dims[spatial_dim].window_dilation()); in SwapConvOperands()
|
| D | shape_inference.cc | 196 if (dim.window_dilation() < 1) { in InferWindowOutputShape()
207 window_util::DilatedBound(dim.size(), dim.window_dilation()); in InferWindowOutputShape()
|
| D | hlo_evaluator_typed_visitor.h | 1189 rhs_spatial_index[ki] * window_dim.window_dilation();
2865 window_index[i] * window.dimensions(i).window_dilation() -
|
| D | sharding_propagation.cc | 801 dimensions.window_dilation() > 1; in InferShardingFromOperands()
|
| D | elemental_ir_emitter.cc | 2653 index_typed_const(window.dimensions(i).window_dilation()))), in EmitElementalReduceWindow()
2878 NSWMul(kernel_index, b_->getInt64(window_dim.window_dilation())); in EmitConvolution()
|
| D | hlo_evaluator_test.cc | 146 int window_dilation, int base_dilation, in ReduceWindowMaxIotaTest() argument
173 dim.set_window_dilation(window_dilation); in ReduceWindowMaxIotaTest()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_backprop_filter_test.cc | 35 int64 output_batch, window, window_dilation; member
69 config.window_dilation = 1; in GetConv2DTestCases()
97 config.window_dilation = 2; in GetConv2DTestCases()
165 spec.window, spec.window_dilation, spec.window_dilation, in BuildHloTextBatchGroupedConvolution2D()
166 spec.window_dilation, spec.window_dilation, spec.output_batch); in BuildHloTextBatchGroupedConvolution2D()
|
| D | reduce_window_test.cc | 1076 int64 window_dilation[2]; member
1220 "__window_dilation_", absl::StrJoin(param.window_dilation, "x"), // in R2ReduceWindowTestDataToString()
1272 /*window_dilations=*/param.window_dilation, in DoIt()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | window_util.cc | 81 if (dim.window_dilation() != 1) { in ToString()
82 StrAppend(&str, ",window_dilation=", dim.window_dilation()); in ToString()
127 return StrCat(dim.window_dilation()); in ToString()
187 if (dim.window_dilation() != 1) { in HasWindowDilation()
221 window_dimension.window_dilation() == 1 && in IsTrivialWindowDimension()
|
| D | xla_data.proto | 444 // of 1 means no dilation. window_dilation - 1 no-op entries ("holes") are
447 int64 window_dilation = 5; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 377 (1 + (wd.size() - 1) * wd.window_dilation())) / in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
381 rhs_shard_size * wd.window_dilation() - lhs_shard_size, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
383 wd.window_dilation() - 1, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
384 wd.window_dilation())); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
387 lhs_shard_size - rhs_shard_size * wd.window_dilation(), in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
388 lhs_shard_size - rhs_shard_size * wd.window_dilation() + in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
389 padding_low + wd.window_dilation() - 1, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
390 wd.window_dilation())); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
414 0, wd.window_dilation(), 1))) - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
416 rhs_shard_size * wd.window_dilation() - lhs_shard_size, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 248 dim->set_stride(conv->window().dimensions(i).window_dilation()); in MatchBackwardFilter()
386 if (window_dim.window_dilation() != 1) { in MatchBackwardInput()
|
| D | gpu_conv_padding_legalization.cc | 138 conv_window.dimensions(i).window_dilation() - 1); in MaybePaddedKernel()
|
| D | gpu_conv_runner.cc | 413 window.dimensions(dim).window_dilation()); in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 663 win_dilations.push_back(dim.window_dilation()); in ImportInstructionImpl()
696 rhs_dilations.push_back(dim.window_dilation()); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 760 return static_cast<int64_t>(dim.window_dilation()); in EmitDnnConvolution()
1202 window_dilations.push_back(dim.window_dilation()); in EmitReduceWindowOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 183 win_dilations.push_back(dim.window_dilation()); in ReduceWindowInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 950 int64 rhs_row_dilation = window.dimensions(0).window_dilation(); in HandleConvolution()
952 one_dim_convolution ? 1 : window.dimensions(1).window_dilation(); in HandleConvolution()
|