| /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 | 1418 std::vector<int64> window_dimensions( in ConvWithGeneralDimensions() local
1420 for (std::vector<int64>::size_type i = 0; i < window_dimensions.size(); in ConvWithGeneralDimensions()
1422 window_dimensions[i] = in ConvWithGeneralDimensions()
1427 MakePadding(base_area_dimensions, window_dimensions, in ConvWithGeneralDimensions()
1462 std::vector<int64> window_dimensions( in ConvGeneralDilated() local
1464 for (std::vector<int64>::size_type i = 0; i < window_dimensions.size(); in ConvGeneralDilated()
1466 window_dimensions[i] = in ConvGeneralDilated()
1472 window_dimensions, window_strides, padding, in ConvGeneralDilated()
1496 std::vector<int64> window_dimensions( in DynamicConvInstruction() local
1498 for (std::vector<int64>::size_type i = 0; i < window_dimensions.size(); ++i) { in DynamicConvInstruction()
[all …]
|
| D | xla_builder.h | 702 absl::Span<const int64> window_dimensions,
708 absl::Span<const int64> window_dimensions,
714 absl::Span<const int64> window_dimensions,
722 absl::Span<const int64> window_dimensions,
763 absl::Span<const int64> window_dimensions,
770 absl::Span<const int64> window_dimensions,
777 absl::Span<const int64> window_dimensions,
1279 absl::Span<const int64> window_dimensions,
1285 absl::Span<const int64> window_dimensions,
1290 absl::Span<const int64> window_dimensions,
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 53 std::vector<int64> window_dimensions; in HandleReduce() local
57 window_dimensions.push_back(1); in HandleReduce()
67 window_dimensions.push_back(reduce_window_size_); in HandleReduce()
72 MakePadding(AsInt64Slice(input_shape.dimensions()), window_dimensions, in HandleReduce()
77 window_dimensions, window_strides, padding, {}, {})); in HandleReduce()
|
| D | shape_inference.h | 338 absl::Span<const int64> window_dimensions,
|
| D | shape_inference.cc | 2349 absl::Span<const int64> window_dimensions, in InferWindowFromDimensions() argument
2355 if (x == 0 || x == window_dimensions.size()) { in InferWindowFromDimensions()
2362 "\nNumber of window dimensions: ", window_dimensions.size(), in InferWindowFromDimensions()
2372 for (size_t i = 0; i < window_dimensions.size(); i++) { in InferWindowFromDimensions()
2374 dim->set_size(window_dimensions[i]); in InferWindowFromDimensions()
|
| D | shape_inference_test.cc | 1109 std::vector<int64> window_dimensions = {1, 2, 4}; in TEST_F() local
1112 MakePadding(AsInt64Slice(f32_arg_shape.dimensions()), window_dimensions, in TEST_F()
1117 window_dimensions, window_strides, padding_values, {}, {})); in TEST_F()
1171 std::vector<int64> window_dimensions = {1, 2, 4}; in TEST_F() local
1174 MakePadding(AsInt64Slice(f32_arg_shape.dimensions()), window_dimensions, in TEST_F()
1179 window_dimensions, window_strides, padding_values, {}, {})); in TEST_F()
|
| /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 | 477 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()
120 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()
126 context->Input("operand"), *select.computation, window_dimensions, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 351 window_dimensions, argument
377 window_strides = window_strides or [1] * len(window_dimensions)
378 base_dilations = base_dilations or [1] * len(window_dimensions)
379 window_dilations = window_dilations or [1] * len(window_dimensions)
380 padding = padding or [(0, 0)] * len(window_dimensions)
384 window_dimensions=window_dimensions,
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 110 absl::Span<const int64> window_dimensions, in AvgPoolDivideByCount() argument
120 std::accumulate(window_dimensions.begin(), window_dimensions.end(), 1, in AvgPoolDivideByCount()
127 padding, window_dimensions, in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 793 const uint64_t rank = rw.window_dimensions().size(); in matchAndRewrite()
821 if (rw.window_dimensions().getValue<int64_t>({batch_dim}) != 1 || in matchAndRewrite()
822 rw.window_dimensions().getValue<int64_t>({channel_dim}) != 1 || in matchAndRewrite()
844 for (int64_t w : rw.window_dimensions().getValues<int64_t>()) { in matchAndRewrite()
856 llvm::to_vector<4>(rw.window_dimensions().getValues<int64_t>()), in matchAndRewrite()
877 if (rw.window_dimensions() != rw_rhs.window_dimensions() || in matchAndRewrite()
887 llvm::to_vector<4>(rw.window_dimensions().getValues<int64_t>()), in matchAndRewrite()
900 rw.window_dimensions().getValue<int64_t>({i}) - in matchAndRewrite()
910 llvm::to_vector<4>(rw.window_dimensions().getValues<int64_t>()), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1584 window_dimensions = (2, 1)
1588 c.get_shape(operand).dimensions(), window_dimensions, window_strides)
1592 window_dimensions=window_dimensions,
1659 window_dimensions = (2, 1)
1662 xla_client.PaddingType.VALID, input_array.shape, window_dimensions,
1668 window_dimensions=window_dimensions,
1684 window_dimensions = (2, 1)
1687 xla_client.PaddingType.SAME, input_array.shape, window_dimensions,
1693 window_dimensions=window_dimensions,
1709 window_dimensions = (2, 1)
[all …]
|
| /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 std::vector<int64> window_dimensions; member
83 SelectAndScatter(operand, ge_f32_, GetParam().window_dimensions, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 400 for (const auto& window_dim : reduce_window_op.window_dimensions()) { in CreateParallelLoopsToTraverseOutputAndWindow()
557 s_and_s_op.window_dimensions()->getIntValues()) { in InsertWindowLoops()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 450 llvm::SmallVector<int64_t, 4> window_strides, window_dimensions; in ImportInstructionImpl() local
454 window_dimensions.push_back(dim.size()); in ImportInstructionImpl()
461 Convert(window_dimensions))); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1323 …// CHECK: {window_dimensions = dense<[1, 2, 2, 1]> : tensor<4xi64>, window_strides = dense<[1, 4, …
1345 …// CHECK: {window_dimensions = dense<[1, 1, 2, 2, 1]> : tensor<5xi64>, window_strides = dense<[1, …
1385 …// CHECK: }) {window_dimensions = dense<[1, 2, 2, 1]> : tensor<4xi64>, window_strides = dense<[1, …
1408 …// CHECK: }) {window_dimensions = dense<[1, 1, 2, 2, 1]> : tensor<5xi64>, window_strides = dense<[…
4410 // CHECK-SAME: window_dimensions = dense<[1, 2, 2, 1]>
4434 // CHECK-SAME: window_dimensions = dense<[1, 1, 2, 2, 1]>
4458 // CHECK-SAME: window_dimensions = dense<[1, 1, 2, 2]>
4482 // CHECK-SAME: window_dimensions = dense<[1, 1, 1, 2, 2]>
4506 // CHECK-SAME: window_dimensions = dense<[1, 5, 2, 1]>
4516 // CHECK-SAME: window_dimensions = dense<[1, 5, 2, 1]>
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo-legalize-select-and-scatter.mlir | 27 window_dimensions = dense<[3, 3]> : tensor<2xi64>,
|
| D | lhlo-legalize-to-parallel-loops.mlir | 145 window_dimensions = dense<[3, 3]> : tensor<2xi64>,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_ops.td | 225 I64ElementsAttr:$window_dimensions,
496 OptionalAttr<I64ElementsAttr>:$window_dimensions,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 78 // CHECK: window_dimensions = dense<3> : tensor<1xi64>
544 // CHECK-SAME: window_dimensions = dense<[1, 2, 2, 1]> : tensor<4xi64>,
|