| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | padding.cc | 28 absl::Span<const int64> window_strides) { in ValidatePaddingValues()
44 absl::Span<const int64> window_strides, Padding padding) { in MakePadding()
|
| D | xla_builder.cc | 1212 absl::Span<const int64> window_strides, Padding padding, in Conv()
1222 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvWithGeneralPadding()
1232 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvWithGeneralDimensions()
1268 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvGeneral()
1279 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvGeneralDilated()
1963 absl::Span<const int64> window_strides, in ReduceWindow()
1983 absl::Span<const int64> window_strides, in ReduceWindowWithGeneralPadding()
2276 absl::Span<const int64> window_strides, in SelectAndScatter()
2293 absl::Span<const int64> window_strides, in SelectAndScatterWithGeneralPadding()
3036 absl::Span<const int64> window_strides, Padding padding, in Conv()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 47 std::vector<int64> window_strides; in Compile() local
|
| D | reduce_window_op.cc | 42 std::vector<int64> window_strides; in Compile() local
|
| D | xla_select_and_scatter_op.cc | 43 std::vector<int64> window_strides; in Compile() local
|
| D | scan_ops.cc | 80 std::vector<int64> window_strides(input_shape.dims(), 1); in Compile() local
|
| D | extract_image_patches_op.cc | 126 std::vector<int64> window_strides(num_spatial_dims); in Compile() local
|
| D | conv_op_helpers.cc | 255 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
421 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 54 std::vector<int64> window_strides; in HandleReduce() local
|
| D | shape_inference.cc | 2258 absl::Span<const int64> window_strides, in InferWindowFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 673 window_strides): argument
1080 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, argument
1250 window_dimensions, window_strides, padding): argument
1274 window_strides, base_dilations, argument
1393 window_strides, argument
1426 window_strides, argument
1478 window_strides, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 238 window_strides, argument
342 window_strides=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 110 absl::Span<const int64> window_strides, in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 74 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()
|
| D | select_and_scatter_test.cc | 46 absl::Span<const int64> window_strides; member
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 754 SmallVector<int64_t, 4> window_strides; in matchAndRewrite() local
1134 ArrayAttr window_strides, StringRef padding, Builder *builder) { in GetReduceWindowPadding()
2660 SmallVector<int64_t, 4> window_strides(num_spatial_dims); in matchAndRewrite() local
|