Home
last modified time | relevance | path

Searched refs:window_util (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util_test.cc28 window_util::HasOverlappingWindow(window_util::MakeWindow({1, 1}))); in TEST()
30 window_util::HasOverlappingWindow(window_util::MakeWindow({2, 2, 2, 2}))); in TEST()
35 Window w = window_util::MakeWindow({1, 2}, {3, 4}); in TEST()
Dwindow_util.h24 namespace window_util {
DBUILD863 name = "window_util",
864 srcs = ["window_util.cc"],
865 hdrs = ["window_util.h"],
882 ":window_util",
898 ":window_util",
Dwindow_util.cc27 namespace window_util { namespace
Dreference_util.cc139 return window_util::StridedBound(unpadded_width, window_len, stride); in WindowCount()
158 window_util::StridedBound(padded_width, window_size, stride_amount); in ReduceWindow1DGeneric()
260 window_util::StridedBound(padded_width, window[i], stride[i]); in ReduceWindow4DGeneric()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc37 return window_util::HasSymmetricPadding(conv.window()) && in IsForwardConvolutionCanonical()
38 !window_util::HasNegativePadding(conv.window()) && in IsForwardConvolutionCanonical()
39 !window_util::HasDilation(conv.window()); in IsForwardConvolutionCanonical()
53 if (!window_util::HasSymmetricPadding(*conv_window) || in MaybePaddedAndSlicedInput()
54 window_util::HasBaseDilation(*conv_window)) { in MaybePaddedAndSlicedInput()
90 if (window_util::HasNegativePadding(*conv_window)) { in MaybePaddedAndSlicedInput()
128 if (!window_util::HasWindowDilation(conv_window)) { in MaybePaddedKernel()
209 if (window_util::HasSymmetricPadding(backward_conv->window())) { in CanonicalizeBackwardFilterConvolution()
280 if (window_util::HasSymmetricPadding(backward_conv->window())) { in CanonicalizeBackwardInputConvolution()
Dgpu_conv_rewriter.cc58 ? !window_util::AllOrNoneReversed(conv->window()) in CanImplementAsGpuForwardConv()
59 : window_util::HasWindowReversal(conv->window())) { in CanImplementAsGpuForwardConv()
151 !window_util::HasWindowDilation(conv->window())) { in MatchBackwardFilter()
318 !(window_util::HasBaseDilation(conv->window()) && in MatchBackwardInput()
Dcudnn_support_utils.cc110 if (window_util::HasDilation(conv.window())) { in CudnnSupportsOptimizedIntegerConvolution()
Dgpu_layout_assignment.cc131 window_util::HasStride(instr->window())) { in HeuristicLayoutAssignment()
DBUILD317 "//tensorflow/compiler/xla:window_util",
385 "//tensorflow/compiler/xla:window_util",
1051 "//tensorflow/compiler/xla:window_util",
1319 "//tensorflow/compiler/xla:window_util",
1355 "//tensorflow/compiler/xla:window_util",
1398 "//tensorflow/compiler/xla:window_util",
1506 "//tensorflow/compiler/xla:window_util",
2018 "//tensorflow/compiler/xla:window_util",
/external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/
DBUILD112 "window_util.c",
116 "window_util.h",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc78 if (window_util::HasWindowReversal(convolution.window())) { in PotentiallyImplementedAsEigenConvolution()
DBUILD518 "//tensorflow/compiler/xla:window_util",
1166 "//tensorflow/compiler/xla:window_util",
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
DBUILD34 "//tensorflow/compiler/xla:window_util",
Dconv_emitter.cc599 if (window_util::HasWindowReversal(conv->window())) { in ConvIsImplemented()
602 if (window_util::HasDilation(conv->window())) { in ConvIsImplemented()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
DBUILD42 "//tensorflow/compiler/xla:window_util",
Dconvolution_handler.cc349 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
780 auto operand_window = window_util::MakeWindow(ones); in PartitionConvolutionTiledOutput()
/external/tensorflow/tensorflow/compiler/xla/service/
DBUILD386 "//tensorflow/compiler/xla:window_util",
408 "//tensorflow/compiler/xla:window_util",
465 "//tensorflow/compiler/xla:window_util",
624 "//tensorflow/compiler/xla:window_util",
951 "//tensorflow/compiler/xla:window_util",
2369 "//tensorflow/compiler/xla:window_util",
2399 "//tensorflow/compiler/xla:window_util",
2432 "//tensorflow/compiler/xla:window_util",
2519 "//tensorflow/compiler/xla:window_util",
2916 "//tensorflow/compiler/xla:window_util",
[all …]
Ddynamic_padder.cc921 const int64_t dilated_window_size = window_util::DilatedBound( in RewriteInputWithDynamicPadding()
943 window_util::DilatedBound(padded_shape.dimensions(shape_dim), in RewriteInputWithDynamicPadding()
1194 if (window_util::IsTrivialWindowDimension(window_dim)) { in RewriteDynamicReduceWindowSamePadding()
1231 if (window_util::IsTrivialWindowDimension(window_dim)) { in RewriteDynamicSelectAndScatterSamePadding()
1278 const int64_t dilated_window_size = window_util::DilatedBound( in RewriteDynamicSelectAndScatterSamePadding()
Dhlo_instructions.cc2442 if (window_util::HasBaseDilation(window)) { in HloConvolutionInstruction()
2445 if (window_util::HasWindowDilation(window)) { in HloConvolutionInstruction()
2454 if (window_util::HasBaseDilation(window())) { in ToCategory()
2457 if (window_util::HasWindowDilation(window())) { in ToCategory()
2478 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2561 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2613 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2755 extra.push_back(StrCat("window={", window_util::ToString(*window_), "}")); in ExtraAttributesToStringImpl()
Dalgebraic_simplifier.cc5742 if (window_util::HasStride(window) || window_util::HasDilation(window) || in HandleReduceWindow()
5743 window_util::HasPadding(window)) { in HandleReduceWindow()
5800 if (HasInteriorPadding(pad_config) && window_util::HasBaseDilation(window)) { in HandleReduceWindow()
5838 if (window_util::HasStride(window)) { in HandleReduceWindow()
5842 if (!window_util::HasSymmetricPadding(pad_config)) { in HandleReduceWindow()
6579 if (window_util::HasStride(window) || window_util::HasPadding(window) || in SimplifyConvToDot()
6580 window_util::HasBaseDilation(window)) { in SimplifyConvToDot()
Ddynamic_dimension_inference.cc780 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleDynamicWindowSamePadding()
1245 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleReduceWindow()
Dshape_inference.cc194 const int64_t dilated_base = window_util::DilatedBound( in InferWindowOutputShape()
199 window_util::DilatedBound(dim.size(), dim.window_dilation()); in InferWindowOutputShape()
201 output_dimensions[i] = window_util::StridedBound( in InferWindowOutputShape()
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD258 "//tensorflow/compiler/xla:window_util",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc1405 const int64_t dilated_base = xla::window_util::DilatedBound( in InferWindowOutputShape()
1410 xla::window_util::DilatedBound(dim.size(), dim.window_dilation()); in InferWindowOutputShape()
1412 output_dimensions[i] = xla::window_util::StridedBound( in InferWindowOutputShape()

12