| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 1446 absl::Span<const int64_t> window_strides, in Conv()
1459 XlaOp lhs, XlaOp rhs, absl::Span<const int64_t> window_strides, in ConvWithGeneralPadding()
1471 XlaOp lhs, XlaOp rhs, absl::Span<const int64_t> window_strides, in ConvWithGeneralDimensions()
1509 XlaOp lhs, XlaOp rhs, absl::Span<const int64_t> window_strides, in ConvGeneral()
1525 XlaOp lhs, XlaOp rhs, absl::Span<const int64_t> window_strides, in ConvGeneralDilated()
1565 XlaOp lhs, XlaOp rhs, absl::Span<const int64_t> window_strides, in DynamicConvInstruction()
1609 absl::Span<const int64_t> window_strides, in DynamicConvInputGrad()
1634 absl::Span<const int64_t> window_strides, in DynamicConvKernelGrad()
1661 XlaOp lhs, XlaOp rhs, absl::Span<const int64_t> window_strides, in DynamicConvForward()
1684 absl::Span<const int64_t> window_strides, in ConvGeneralDilatedInternal()
[all …]
|
| H A D | padding.cc | 28 absl::Span<const int64_t> window_strides) { in ValidatePaddingValues()
44 absl::Span<const int64_t> window_strides, Padding padding) { in MakePadding()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| H A D | replace_cast_hacks_with_tf_xla_ops.cc | 57 Value &window_strides, Value &lhs_dilation, in PrepareXlaConvParams()
130 Value padding, window_strides, lhs_dilation, rhs_dilation, in CreateXLAConvOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 1312 llvm::SmallVector<int64_t> window_strides, in InferWindowFromDimensions()
1426 DenseElementsAttr window_dimensions, window_strides, base_dilations, in InferShapeForXlaReduceWindowOp() local
1491 DenseElementsAttr window_dimensions, window_strides, padding; in InferShapeForXlaSelectAndScatterOp() local
1541 llvm::SmallVector<int64_t> window_strides, in InferXlaConvOutputShape()
1597 auto window_strides = op.window_strides(); in PrecheckForXlaConvV2Op() local
1730 auto window_strides = op.window_strides(); in InferShapeForXlaConvV2Op() local
|
| H A D | legalize_hlo.cc | 936 auto window_strides = conv_op.window_strides().getValue(); in MatchResizeOp() local
2293 mhlo::ReduceWindowOp rw, llvm::SmallVectorImpl<int64_t> *window_strides, in IsSpatialPoolingWithoutDilation()
2538 llvm::SmallVector<int64_t, 5> window_strides; in matchAndRewrite() local
2652 llvm::SmallVector<int64_t, 5> window_strides; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | conv_op_helpers.cc | 234 std::vector<int64_t> window_strides(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
439 std::vector<int64_t> window_strides(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
|
| H A D | scan_ops.cc | 78 std::vector<int64_t> window_strides(input_shape.dims(), 1); in Compile() local
|
| H A D | extract_image_patches_op.cc | 127 std::vector<int64_t> window_strides(num_spatial_dims); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | tree_reduction_rewriter.cc | 70 std::vector<int64_t> window_strides; in HandleReduce() local
|
| H A D | shape_inference_test.cc | 1145 std::vector<int64_t> window_strides = {1, 1, 1}; in TEST_F() local
1207 std::vector<int64_t> window_strides = {1, 1, 1}; in TEST_F() local
|
| H A D | shape_inference.cc | 2457 absl::Span<const int64_t> window_strides, in InferWindowFromDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 245 window_strides, argument
420 window_strides=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reduce_window_test.cc | 72 absl::Span<const int64_t> window_strides, in ReduceWindowAdd()
83 absl::Span<const int64_t> window_strides, in ReduceWindowMax()
94 absl::Span<const int64_t> window_strides, in ReduceWindowMin()
|
| H A D | select_and_scatter_test.cc | 48 std::vector<int64_t> window_strides; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 1150 SmallVector<int64_t, num_spatial_dims> window_strides; in matchAndRewriteDynamicConv() local
1314 SmallVector<int64_t, num_spatial_dims> window_strides; in matchAndRewrite() local
2538 ArrayAttr window_strides, StringRef padding, Builder *builder) { in GetReduceWindowPaddingAsArray()
2564 ArrayAttr window_strides, StringRef padding, Builder *builder) { in GetReduceWindowPaddingAsAttr()
5172 SmallVector<int64_t, num_spatial_dims> window_strides; in matchAndRewrite() local
6161 DenseElementsAttr window_dimensions, window_strides, base_dilations, in matchAndRewrite() local
6314 SmallVector<int64_t, 4> window_strides(rank, 1); in matchAndRewrite() local
7071 ElementsAttr window_dimensions, window_strides, padding; in matchAndRewrite() local
|
| H A D | mhlo_to_lhlo_with_xla.cc | 937 DenseIntElementsAttr window_strides; in EmitDnnConvolution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | pooling.cc | 112 absl::Span<const int64_t> window_strides, in AvgPoolDivideByCount()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | xla_client.py | 358 window_strides): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 102 absl::Span<const int64_t> window_strides, in ConvGeneralDilatedInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | hlo_function_importer.cc | 823 llvm::SmallVector<int64_t, 4> window_strides, window_dimensions; in ImportInstructionImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | jitrt_custom_calls.cc | 690 ArrayRef<int64_t> window_strides; member
|
| H A D | ir_emitter_unnested.cc | 1016 mlir::DenseIntElementsAttr window_strides = in EmitConvolutionThunk() local
|