| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | padding.cc | 27 absl::Span<const int64> window_dimensions, in ValidatePaddingValues() argument
29 bool ok = input_dimensions.size() == window_dimensions.size() && in ValidatePaddingValues()
35 input_dimensions.size(), window_dimensions.size(), in ValidatePaddingValues()
43 absl::Span<const int64> window_dimensions, in MakePadding() argument
45 TF_CHECK_OK(ValidatePaddingValues(input_dimensions, window_dimensions, in MakePadding()
50 low_high_padding.resize(window_dimensions.size(), {0, 0}); in MakePadding()
56 int64 window_dimension = window_dimensions[i]; in MakePadding()
|
| D | padding.h | 45 absl::Span<const int64> window_dimensions,
61 absl::Span<const int64> window_dimensions,
|
| D | xla_builder.cc | 1139 std::vector<int64> window_dimensions( in ConvWithGeneralDimensions() local
1141 for (std::vector<int64>::size_type i = 0; i < window_dimensions.size(); in ConvWithGeneralDimensions()
1143 window_dimensions[i] = in ConvWithGeneralDimensions()
1148 MakePadding(base_area_dimensions, window_dimensions, in ConvWithGeneralDimensions()
1180 std::vector<int64> window_dimensions( in ConvGeneralDilated() local
1182 for (std::vector<int64>::size_type i = 0; i < window_dimensions.size(); in ConvGeneralDilated()
1184 window_dimensions[i] = in ConvGeneralDilated()
1188 MakeWindow(window_dimensions, window_strides, padding, in ConvGeneralDilated()
1211 absl::Span<const int64> window_dimensions, in MakeWindow() argument
1217 if (x == 0 || x == window_dimensions.size()) { in MakeWindow()
[all …]
|
| D | xla_builder.h | 454 absl::Span<const int64> window_dimensions,
460 absl::Span<const int64> window_dimensions,
485 absl::Span<const int64> window_dimensions,
493 absl::Span<const int64> window_dimensions,
648 StatusOr<Window> MakeWindow(absl::Span<const int64> window_dimensions,
873 absl::Span<const int64> window_dimensions,
879 absl::Span<const int64> window_dimensions,
899 absl::Span<const int64> window_dimensions,
906 absl::Span<const int64> window_dimensions,
1550 absl::Span<const int64> window_dimensions,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_window_test.py | 61 window_dimensions=[2]))
69 window_dimensions=[2],
78 window_dimensions=[1],
87 window_dimensions=[2, 2],
96 window_dimensions=[2, 2],
|
| D | xla_ops_test.py | 274 window_dimensions=[2, 3, 1, 1],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduce_window_op.cc | 41 std::vector<int64> window_dimensions; in Compile() local
46 "window_dimensions", &window_dimensions)); in Compile()
55 OP_REQUIRES(context, rank == window_dimensions.size(), in Compile()
59 window_dimensions.size(), " vs. ", rank, ")")); in Compile()
121 window_dimensions, window_strides, base_dilations, window_dilations, in Compile()
|
| D | xla_select_and_scatter_op.cc | 42 std::vector<int64> window_dimensions; in Compile() local
45 "window_dimensions", &window_dimensions)); in Compile()
50 OP_REQUIRES(context, rank == window_dimensions.size(), in Compile()
54 window_dimensions.size(), " vs. ", rank, ")")); in Compile()
127 context->Input("operand"), *select.computation, window_dimensions, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 353 window_dimensions, argument
379 window_strides = window_strides or [1] * len(window_dimensions)
380 base_dilations = base_dilations or [1] * len(window_dimensions)
381 window_dilations = window_dilations or [1] * len(window_dimensions)
382 padding = padding or [(0, 0)] * len(window_dimensions)
386 window_dimensions=window_dimensions,
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 109 absl::Span<const int64> window_dimensions, in AvgPoolDivideByCount() argument
119 std::accumulate(window_dimensions.begin(), window_dimensions.end(), 1, in AvgPoolDivideByCount()
126 padding, window_dimensions, in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1282 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, argument
1303 padding, self.GetShape(operand).dimensions(), window_dimensions,
1306 operand, select.computation, window_dimensions, window_strides, pads,
1473 window_dimensions, window_strides, padding): argument
1489 self.GetShape(operand).dimensions(), window_dimensions, window_strides)
1492 window_dimensions, window_strides, (), (), pads)
1495 self, operand, init_value, computation_to_apply, window_dimensions, argument
1514 window_dimensions, window_strides, base_dilations, window_dilations,
|
| D | xla_client_test.py | 1434 window_dimensions=(2, 1),
1446 window_dimensions=(2, 1),
1557 window_dimensions=(2, 1), window_strides=(1, 1),
1567 window_dimensions=(2, 1), window_strides=(1, 1),
1577 window_dimensions=(2, 1), window_strides=(1, 2),
1587 window_dimensions=(2, 1), window_strides=(1, 1),
1597 window_dimensions=(2, 1), window_strides=(1, 1),
1607 window_dimensions=(2, 1), window_strides=(1, 2),
|
| D | local_computation_builder.cc | 552 absl::Span<const int64> window_dimensions, in SelectAndScatterWithGeneralPadding() argument
557 operand.op(), select.computation(), window_dimensions, window_strides, in SelectAndScatterWithGeneralPadding()
667 absl::Span<const int64> window_dimensions, in ReduceWindowWithGeneralPadding() argument
674 window_dimensions, window_strides, base_dilations, window_dilations, in ReduceWindowWithGeneralPadding()
|
| D | local_computation_builder.h | 279 absl::Span<const int64> window_dimensions,
333 absl::Span<const int64> window_dimensions,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 73 absl::Span<const int64> window_dimensions, in ReduceWindowAdd() argument
80 window_dimensions, window_strides, padding); in ReduceWindowAdd()
84 absl::Span<const int64> window_dimensions, in ReduceWindowMax() argument
91 window_dimensions, window_strides, padding); in ReduceWindowMax()
95 absl::Span<const int64> window_dimensions, in ReduceWindowMin() argument
102 window_dimensions, window_strides, padding); in ReduceWindowMin()
|
| D | select_and_scatter_test.cc | 45 absl::Span<const int64> window_dimensions; member
83 SelectAndScatter(operand, ge_f32_, GetParam().window_dimensions, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1870 <b> `ReduceWindow(operand, init_value, computation, window_dimensions,
1885 | `window_dimensions` | `ArraySlice<int64>` | array of integers for window |
1898 matrix of size [4x6] and both window_dimensions and window_stride_dimensions are
1920 /*window_dimensions=*/{2, 3},
1932 window_dimensions. The figure below illustrates the use of two different stride
2300 <b>`SelectAndScatter(operand, select, window_dimensions, window_strides,
2313 | `window_dimensions` | `ArraySlice<int64>` | array of integers for window |
|