Home
last modified time | relevance | path

Searched defs:window_strides (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc28 absl::Span<const int64> window_strides) { in ValidatePaddingValues()
44 absl::Span<const int64> window_strides, Padding padding) { in MakePadding()
Dxla_builder.cc1100 absl::Span<const int64> window_strides, Padding padding, in Conv()
1110 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvWithGeneralPadding()
1120 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvWithGeneralDimensions()
1156 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvGeneral()
1167 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvGeneralDilated()
1212 absl::Span<const int64> window_strides, in MakeWindow()
1996 absl::Span<const int64> window_strides, in ReduceWindow()
2019 absl::Span<const int64> window_strides, in ReduceWindowWithGeneralPadding()
2247 absl::Span<const int64> window_strides, in SelectAndScatter()
2264 absl::Span<const int64> window_strides, in SelectAndScatterWithGeneralPadding()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc47 std::vector<int64> window_strides; in Compile() local
Dreduce_window_op.cc42 std::vector<int64> window_strides; in Compile() local
Dxla_select_and_scatter_op.cc43 std::vector<int64> window_strides; in Compile() local
Dscan_ops.cc80 std::vector<int64> window_strides(input_shape.dims(), 1); in Compile() local
Dextract_image_patches_op.cc121 std::vector<int64> window_strides(num_spatial_dims); in Compile() local
Dconv_op_helpers.cc286 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
446 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py230 window_strides, argument
354 window_strides=None, argument
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py251 window_strides): argument
1282 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, argument
1473 window_dimensions, window_strides, padding): argument
1496 window_strides, base_dilations, window_dilations, padding): argument
1620 def Conv(self, lhs, rhs, window_strides, padding, feature_group_count=1): argument
1639 def ConvWithGeneralPadding(self, lhs, rhs, window_strides, padding, argument
1674 def ConvGeneralDilated(self, lhs, rhs, window_strides, padding, lhs_dilation, argument
Dlocal_computation_builder.cc553 absl::Span<const int64> window_strides, in SelectAndScatterWithGeneralPadding()
588 absl::Span<const int64> window_strides, in ConvGeneralDilated()
668 absl::Span<const int64> window_strides, in ReduceWindowWithGeneralPadding()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc110 absl::Span<const int64> window_strides, in AvgPoolDivideByCount()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc74 absl::Span<const int64> window_strides, in ReduceWindowAdd()
85 absl::Span<const int64> window_strides, in ReduceWindowMax()
96 absl::Span<const int64> window_strides, in ReduceWindowMin()
Dselect_and_scatter_test.cc46 absl::Span<const int64> window_strides; member