| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | padding.cc | 28 absl::Span<const int64> window_strides) { in ValidatePaddingValues() argument
30 input_dimensions.size() == window_strides.size(); in ValidatePaddingValues()
36 window_strides.size()); in ValidatePaddingValues()
44 absl::Span<const int64> window_strides, Padding padding) { in MakePadding() argument
46 window_strides)); in MakePadding()
57 int64 window_stride = window_strides[i]; in MakePadding()
|
| D | xla_builder.h | 384 absl::Span<const int64> window_strides, Padding padding,
390 absl::Span<const int64> window_strides,
397 absl::Span<const int64> window_strides, Padding padding,
403 absl::Span<const int64> window_strides,
410 absl::Span<const int64> window_strides,
455 absl::Span<const int64> window_strides, Padding padding);
461 absl::Span<const int64> window_strides,
486 absl::Span<const int64> window_strides,
494 absl::Span<const int64> window_strides,
649 absl::Span<const int64> window_strides,
[all …]
|
| D | padding.h | 46 absl::Span<const int64> window_strides);
62 absl::Span<const int64> window_strides, Padding padding);
|
| D | xla_builder.cc | 1100 absl::Span<const int64> window_strides, Padding padding, in Conv() argument
1104 lhs, rhs, window_strides, padding, in Conv()
1105 CreateDefaultConvDimensionNumbers(window_strides.size()), in Conv()
1110 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvWithGeneralPadding() argument
1114 return ConvGeneral(lhs, rhs, window_strides, padding, in ConvWithGeneralPadding()
1115 CreateDefaultConvDimensionNumbers(window_strides.size()), in ConvWithGeneralPadding()
1120 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvWithGeneralDimensions() argument
1147 return ConvGeneral(lhs, rhs, window_strides, in ConvWithGeneralDimensions()
1149 window_strides, padding), in ConvWithGeneralDimensions()
1156 const XlaOp& lhs, const XlaOp& rhs, absl::Span<const int64> window_strides, in ConvGeneral() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduce_window_op.cc | 42 std::vector<int64> window_strides; in Compile() local
48 &window_strides)); in Compile()
60 OP_REQUIRES(context, rank == window_strides.size(), in Compile()
64 window_strides.size(), " vs. ", rank, ")")); in Compile()
121 window_dimensions, window_strides, base_dilations, window_dilations, in Compile()
|
| D | xla_select_and_scatter_op.cc | 43 std::vector<int64> window_strides; in Compile() local
47 &window_strides)); in Compile()
55 OP_REQUIRES(context, rank == window_strides.size(), in Compile()
59 window_strides.size(), " vs. ", rank, ")")); in Compile()
128 window_strides, padding, context->Input("source"), in Compile()
|
| D | xla_conv_op.cc | 47 std::vector<int64> window_strides; in Compile() local
52 &window_strides)); in Compile()
78 context->Input(0), context->Input(1), window_strides, padding, in Compile()
|
| D | extract_image_patches_op.cc | 121 std::vector<int64> window_strides(num_spatial_dims); in Compile() local
138 window_strides[i] = strides_.at(dim); in Compile()
145 window_strides[i], padding_, &unused_output_size, in Compile()
150 xla::ConvGeneralDilated(ctx->Input(0), filter, window_strides, padding, in Compile()
|
| D | conv_op_helpers.cc | 286 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
303 window_strides[i] = attrs.strides.at(dim); in MakeXlaForwardConvOp()
314 rhs_dilation[i], window_strides[i], attrs.padding, &unused_output_size, in MakeXlaForwardConvOp()
319 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
446 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
478 window_strides[i] = attrs.dilations[dim]; in MakeXlaBackpropFilterConvOp()
533 activations, gradients, window_strides, padding, /*lhs_dilation=*/ones, in MakeXlaBackpropFilterConvOp()
|
| D | scan_ops.cc | 80 std::vector<int64> window_strides(input_shape.dims(), 1); in Compile() local
106 window_dims, window_strides, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reduce_window_test.py | 70 window_strides=[2]))
79 window_strides=[3]))
88 window_strides=[1, 1]))
97 window_strides=[2, 2],
|
| D | xla_ops_test.py | 129 window_strides=(1,),
275 window_strides=[2, 2, 1, 1],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 230 window_strides, argument
265 window_strides=window_strides,
354 window_strides=None, argument
379 window_strides = window_strides or [1] * len(window_dimensions)
387 window_strides=window_strides,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 251 window_strides): argument
267 return [(0, 0)] * len(window_strides)
269 out_shape = np.ceil(np.true_divide(lhs_dims, window_strides)).astype(int)
272 in zip(out_shape, window_strides, rhs_dims, lhs_dims)]
1282 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, argument
1304 window_strides)
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)
[all …]
|
| D | local_computation_builder.cc | 553 absl::Span<const int64> window_strides, in SelectAndScatterWithGeneralPadding() argument
557 operand.op(), select.computation(), window_dimensions, window_strides, in SelectAndScatterWithGeneralPadding()
588 absl::Span<const int64> window_strides, in ConvGeneralDilated() argument
593 return xla::ConvGeneralDilated(lhs.op(), rhs.op(), window_strides, padding, in ConvGeneralDilated()
668 absl::Span<const int64> window_strides, in ReduceWindowWithGeneralPadding() argument
674 window_dimensions, window_strides, base_dilations, window_dilations, in ReduceWindowWithGeneralPadding()
|
| D | local_computation_builder.h | 280 absl::Span<const int64> window_strides,
295 absl::Span<const int64> window_strides,
334 absl::Span<const int64> window_strides,
|
| D | xla_client_test.py | 1435 window_strides=(1, 2),
1447 window_strides=(1, 2),
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),
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 74 absl::Span<const int64> window_strides, in ReduceWindowAdd() argument
80 window_dimensions, window_strides, padding); in ReduceWindowAdd()
85 absl::Span<const int64> window_strides, in ReduceWindowMax() argument
91 window_dimensions, window_strides, padding); in ReduceWindowMax()
96 absl::Span<const int64> window_strides, in ReduceWindowMin() argument
102 window_dimensions, window_strides, padding); in ReduceWindowMin()
|
| D | select_and_scatter_test.cc | 46 absl::Span<const int64> window_strides; member
84 GetParam().window_strides, GetParam().padding_type, source, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 110 absl::Span<const int64> window_strides, in AvgPoolDivideByCount() argument
127 window_strides, data_format); in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 611 | `window_strides` | `ArraySlice<int64>` | n-d array of kernel |
645 The `window_strides` argument specifies the stride of the convolutional window
1871 window_strides, padding)` </b>
1887 | `window_strides` | `ArraySlice<int64>` | array of integers for window |
2300 <b>`SelectAndScatter(operand, select, window_dimensions, window_strides,
2315 | `window_strides` | `ArraySlice<int64>` | array of integers for window |
|