| /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 | padding.h | 46 absl::Span<const int64> window_strides);
62 absl::Span<const int64> window_strides, Padding padding);
|
| D | xla_builder.h | 534 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
541 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
548 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
555 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
563 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
573 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
584 absl::Span<const int64> window_strides,
595 absl::Span<const int64> window_strides,
605 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides,
616 absl::Span<const int64> window_strides,
[all …]
|
| D | xla_builder.cc | 1374 absl::Span<const int64> window_strides, Padding padding, in Conv() argument
1379 lhs, rhs, window_strides, padding, in Conv()
1380 CreateDefaultConvDimensionNumbers(window_strides.size()), in Conv()
1386 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvWithGeneralPadding() argument
1391 return ConvGeneral(lhs, rhs, window_strides, padding, in ConvWithGeneralPadding()
1392 CreateDefaultConvDimensionNumbers(window_strides.size()), in ConvWithGeneralPadding()
1398 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvWithGeneralDimensions() argument
1426 return ConvGeneral(lhs, rhs, window_strides, in ConvWithGeneralDimensions()
1428 window_strides, padding), in ConvWithGeneralDimensions()
1436 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvGeneral() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 54 std::vector<int64> window_strides; in HandleReduce() local
58 window_strides.push_back(1); in HandleReduce()
68 window_strides.push_back(reduce_window_size_); in HandleReduce()
73 window_strides, Padding::kSame); in HandleReduce()
77 window_dimensions, window_strides, padding, {}, {})); in HandleReduce()
|
| D | shape_inference.h | 339 absl::Span<const int64> window_strides,
|
| /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()
120 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()
127 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 | 127 std::vector<int64> window_strides(num_spatial_dims); in Compile() local
144 window_strides[i] = strides_.at(dim); in Compile()
151 window_strides[i], padding_, &unused_output_size, in Compile()
156 xla::ConvGeneralDilated(ctx->Input(0), filter, window_strides, padding, in Compile()
|
| D | conv_op_helpers.cc | 235 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
262 window_strides[i] = attrs.strides.at(dim); in MakeXlaForwardConvOp()
273 rhs_dilation[i], window_strides[i], attrs.padding, &unused_output_size, in MakeXlaForwardConvOp()
279 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
287 conv_input, filter, window_strides, padding, lhs_dilation, rhs_dilation, in MakeXlaForwardConvOp()
439 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
475 window_strides[i] = attrs.dilations[dim]; in MakeXlaBackpropFilterConvOp()
530 activations, gradients, window_strides, padding, /*lhs_dilation=*/ones, in MakeXlaBackpropFilterConvOp()
537 activations, gradients, window_strides, padding, /*lhs_dilation=*/ones, in MakeXlaBackpropFilterConvOp()
|
| D | scan_ops.cc | 78 std::vector<int64> window_strides(input_shape.dims(), 1); in Compile() local
104 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],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 245 window_strides, argument
280 window_strides=window_strides,
352 window_strides=None, argument
377 window_strides = window_strides or [1] * len(window_dimensions)
385 window_strides=window_strides,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 75 conv_op.window_strides().getValue().getValues<int64_t>()) { in matchAndRewrite()
211 if (!conv_op.window_strides().hasValue() || conv_op.window_strides() in IsSupportedConvOp()
799 llvm::SmallVector<int64_t, 5> window_strides; in matchAndRewrite() local
800 if (rw.window_strides().hasValue()) { in matchAndRewrite()
801 window_strides.insert(window_strides.end(), in matchAndRewrite()
802 rw.window_strides()->getValues<int64_t>().begin(), in matchAndRewrite()
803 rw.window_strides()->getValues<int64_t>().end()); in matchAndRewrite()
805 window_strides.resize(rank, 1); in matchAndRewrite()
823 window_strides[batch_dim] != 1 || window_strides[channel_dim] != 1 || in matchAndRewrite()
857 window_strides, "VALID", rewriter); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 46 window_strides = dense<[1, 1]> : tensor<2xi64>,
87 window_strides = dense<1> : tensor<2xi64>}
117 window_strides = dense<1> : tensor<2xi64>,
149 window_strides = dense<1> : tensor<2xi64>}
181 window_strides = dense<1> : tensor<2xi64>}
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 390 window_strides): argument
406 return [(0, 0)] * len(window_strides)
408 out_shape = np.ceil(np.true_divide(lhs_dims, window_strides)).astype(int)
412 out_shape, window_strides, rhs_dims, lhs_dims)
|
| D | xla_client_test.py | 1585 window_strides = (1, 2)
1588 c.get_shape(operand).dimensions(), window_dimensions, window_strides)
1593 window_strides=window_strides,
1660 window_strides = (1, 1)
1663 window_strides)
1669 window_strides=window_strides,
1685 window_strides = (1, 1)
1688 window_strides)
1694 window_strides=window_strides,
1710 window_strides = (1, 2)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | layouts_and_names.mlir | 27 window_strides = dense<2> : tensor<2xi64>
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 97 if (!op.window_strides().hasValue()) { in MapWindowIvsToInput()
100 auto window_strides = op.window_strides().getValue(); in MapWindowIvsToInput() local
115 auto stride = window_strides.template getValue<llvm::APInt>(i); in MapWindowIvsToInput()
|
| /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 std::vector<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 | 111 absl::Span<const int64> window_strides, in AvgPoolDivideByCount() argument
128 window_strides, data_format); in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 79 // CHECK: window_strides = dense<3> : tensor<1xi64>
299 // CHECK-SAME: window_strides = dense<1> : tensor<2xi64>
337 // CHECK-SAME: window_strides = dense<1> : tensor<2xi64>
376 // CHECK-SAME: window_strides = dense<1> : tensor<2xi64>
545 // CHECK-SAME: window_strides = dense<[1, 4, 4, 1]> : tensor<4xi64>}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 115 absl::Span<const int64> window_strides,
|