| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | padding.cc | 27 absl::Span<const int64_t> window_dimensions, in ValidatePaddingValues()
43 absl::Span<const int64_t> window_dimensions, in MakePadding()
|
| H A D | xla_builder.cc | 1491 std::vector<int64_t> window_dimensions( in ConvWithGeneralDimensions() local
1540 std::vector<int64_t> window_dimensions( in ConvGeneralDilated() local
1575 std::vector<int64_t> window_dimensions( in DynamicConvInstruction() local
2644 absl::Span<const int64_t> window_dimensions, in ReduceWindow()
2655 absl::Span<const int64_t> window_dimensions, in ReduceWindow()
2706 absl::Span<const int64_t> window_dimensions, in ReduceWindowWithGeneralPadding()
2751 absl::Span<const int64_t> window_dimensions, in ReduceWindowInternal()
3257 absl::Span<const int64_t> window_dimensions, in SelectAndScatter()
3303 absl::Span<const int64_t> window_dimensions, in SelectAndScatterInternal()
3333 absl::Span<const int64_t> window_dimensions, in SelectAndScatterWithGeneralPadding()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | tree_reduction_rewriter.cc | 69 std::vector<int64_t> window_dimensions; in HandleReduce() local
|
| H A D | shape_inference_test.cc | 1144 std::vector<int64_t> window_dimensions = {1, 2, 4}; in TEST_F() local
1206 std::vector<int64_t> window_dimensions = {1, 2, 4}; in TEST_F() local
|
| H A D | shape_inference.cc | 2456 absl::Span<const int64_t> window_dimensions, in InferWindowFromDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reduce_window_test.cc | 71 absl::Span<const int64_t> window_dimensions, in ReduceWindowAdd()
82 absl::Span<const int64_t> window_dimensions, in ReduceWindowMax()
93 absl::Span<const int64_t> window_dimensions, in ReduceWindowMin()
|
| H A D | select_and_scatter_test.cc | 47 std::vector<int64_t> window_dimensions; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 1311 llvm::SmallVector<int64_t> window_dimensions, in InferWindowFromDimensions()
1426 DenseElementsAttr window_dimensions, window_strides, base_dilations, in InferShapeForXlaReduceWindowOp() local
1491 DenseElementsAttr window_dimensions, window_strides, padding; in InferShapeForXlaSelectAndScatterOp() local
|
| H A D | legalize_hlo.cc | 2424 const auto &window_dimensions = rw.window_dimensions(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | pooling.cc | 111 absl::Span<const int64_t> window_dimensions, in AvgPoolDivideByCount()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 419 window_dimensions, argument
|
| /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/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 6161 DenseElementsAttr window_dimensions, window_strides, base_dilations, in matchAndRewrite() local
7071 ElementsAttr window_dimensions, window_strides, padding; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 2000 mlir::DenseIntElementsAttr window_dimensions = in EmitSelectAndScatter() local
|