| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduce_window_op.cc | 44 std::vector<int64> window_dilations; in Compile() local
52 "window_dilations", &window_dilations)); in Compile()
70 OP_REQUIRES(context, rank == window_dilations.size(), in Compile()
74 window_dilations.size(), " vs. ", rank, ")")); in Compile()
120 window_dimensions, window_strides, base_dilations, window_dilations, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 354 window_dilations=None, argument
379 window_dilations = window_dilations or [1] * len(window_dimensions)
387 window_dilations=window_dilations,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 828 if (rw.window_dilations().hasValue() && in matchAndRewrite()
829 !(rw.window_dilations()->isSplat() && in matchAndRewrite()
830 rw.window_dilations()->getSplatValue<APInt>() == 1)) in matchAndRewrite()
879 rw.window_dilations() != rw_rhs.window_dilations() || in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1196 std::vector<xla::int64> dims, strides, base_dilations, window_dilations, in EmitReduceWindowOp() local
1202 window_dilations.push_back(dim.window_dilation()); in EmitReduceWindowOp()
1215 builder_.getI64TensorAttr(window_dilations)); in EmitReduceWindowOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 717 absl::Span<const int64> window_dilations,
725 absl::Span<const int64> window_dilations,
1293 absl::Span<const int64> window_dilations,
2179 absl::Span<const int64> window_dilations,
|
| D | xla_builder.cc | 2555 absl::Span<const int64> window_dilations, in ReduceWindowWithGeneralPadding() argument
2573 /*rhs_dilation=*/window_dilations)); in ReduceWindowWithGeneralPadding()
2586 window_dilations, padding)); in ReduceWindowWithGeneralPadding()
2600 absl::Span<const int64> window_dilations, in ReduceWindowInternal() argument
2617 /*rhs_dilation=*/window_dilations)); in ReduceWindowInternal()
4645 absl::Span<const int64> window_dilations, in ReduceWindowWithGeneralPadding() argument
4649 window_dimensions, window_strides, base_dilations, window_dilations, in ReduceWindowWithGeneralPadding()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 422 reduce_window_op.window_dilations().hasValue()) { in CreateReduceOpInNestedParallelLoops()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | legalize_hlo.mlir | 1755 window_dilations = dense<1> : tensor<4xi64>,
1777 window_dilations = dense<1> : tensor<4xi64>,
1787 window_dilations = dense<1> : tensor<4xi64>,
1809 window_dilations = dense<1> : tensor<5xi64>,
1831 window_dilations = dense<1> : tensor<4xi64>,
1841 window_dilations = dense<1> : tensor<4xi64>,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1671 window_dilations=[],
1696 window_dilations=[],
1721 window_dilations=[],
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 231 OptionalAttr<I64ElementsAttr>:$window_dilations,
|
| D | hlo_ops.td | 1256 OptionalAttr<I64ElementsAttr>:$window_dilations,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 543 // CHECK-SAME: window_dilations = dense<[1, 2, 2, 1]> : tensor<4xi64>,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 888 ConvertDenseIntAttr(op.window_dilations()), in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 683 window_dilations = dense<[1, 2, 2, 1]> : tensor<4xi64>
|
| D | import.hlotxt | 710 // CHECK-SAME: window_dilations = dense<[1, 2, 2, 1]> : tensor<4xi64>
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2120 | `window_dilations` | `ArraySlice<int64>` | array of integers for window |
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17950 TF_I32OrI64Tensor:$window_dilations,
|