Home
last modified time | relevance | path

Searched refs:window_util (Results 1 – 25 of 36) 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 {
DBUILD817 name = "window_util",
818 srcs = ["window_util.cc"],
819 hdrs = ["window_util.h"],
836 ":window_util",
852 ":window_util",
Dwindow_util.cc27 namespace window_util { namespace
Dreference_util.cc138 return window_util::StridedBound(unpadded_width, window_len, stride); in WindowCount()
157 window_util::StridedBound(padded_width, window_size, stride_amount); in ReduceWindow1DGeneric()
258 window_util::StridedBound(padded_width, window[i], stride[i]); in ReduceWindow4DGeneric()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc36 return window_util::HasSymmetricPadding(conv.window()) && in IsForwardConvolutionCanonical()
37 !window_util::HasNegativePadding(conv.window()) && in IsForwardConvolutionCanonical()
38 !window_util::HasDilation(conv.window()); in IsForwardConvolutionCanonical()
52 if (!window_util::HasSymmetricPadding(*conv_window) || in MaybePaddedAndSlicedInput()
53 window_util::HasBaseDilation(*conv_window)) { in MaybePaddedAndSlicedInput()
88 if (window_util::HasNegativePadding(*conv_window)) { in MaybePaddedAndSlicedInput()
126 if (!window_util::HasWindowDilation(conv_window)) { in MaybePaddedKernel()
206 if (window_util::HasSymmetricPadding(backward_conv->window())) { in CanonicalizeBackwardFilterConvolution()
277 if (window_util::HasSymmetricPadding(backward_conv->window())) { in CanonicalizeBackwardInputConvolution()
Dgpu_conv_rewriter.cc137 ? !window_util::AllOrNoneReversed(conv->window()) in CanImplementAsGpuForwardConv()
138 : window_util::HasWindowReversal(conv->window())) { in CanImplementAsGpuForwardConv()
230 !window_util::HasWindowDilation(conv->window())) { in MatchBackwardFilter()
370 !(window_util::HasBaseDilation(conv->window()) && in MatchBackwardInput()
DBUILD282 "//tensorflow/compiler/xla:window_util",
351 "//tensorflow/compiler/xla:window_util",
687 "//tensorflow/compiler/xla:window_util",
855 "//tensorflow/compiler/xla:window_util",
1106 "//tensorflow/compiler/xla:window_util",
1143 "//tensorflow/compiler/xla:window_util",
1173 "//tensorflow/compiler/xla:window_util",
1512 "//tensorflow/compiler/xla:window_util",
2023 "//tensorflow/compiler/xla:window_util",
Dgpu_layout_assignment.cc107 window_util::HasStride(instr->window())) { in HeuristicLayoutAssignment()
/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.cc79 if (window_util::HasWindowReversal(convolution.window())) { in PotentiallyImplementedAsEigenConvolution()
DBUILD387 "//tensorflow/compiler/xla:window_util",
919 "//tensorflow/compiler/xla:window_util",
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
DBUILD34 "//tensorflow/compiler/xla:window_util",
Dconv_emitter.cc596 if (window_util::HasWindowReversal(conv->window())) { in ConvIsImplemented()
599 if (window_util::HasDilation(conv->window())) { in ConvIsImplemented()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
DBUILD40 "//tensorflow/compiler/xla:window_util",
Dconvolution_handler.cc351 PaddingConfig pad_config = window_util::MakeSymmetricPadding(zero_padding); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
778 auto operand_window = window_util::MakeWindow(ones); in PartitionConvolutionTiledOutput()
/external/tensorflow/tensorflow/compiler/xla/service/
DBUILD224 "//tensorflow/compiler/xla:window_util",
246 "//tensorflow/compiler/xla:window_util",
302 "//tensorflow/compiler/xla:window_util",
459 "//tensorflow/compiler/xla:window_util",
755 "//tensorflow/compiler/xla:window_util",
2046 "//tensorflow/compiler/xla:window_util",
2079 "//tensorflow/compiler/xla:window_util",
2114 "//tensorflow/compiler/xla:window_util",
2145 "//tensorflow/compiler/xla:window_util",
2181 "//tensorflow/compiler/xla:window_util",
[all …]
Ddynamic_padder.cc768 const int64 dilated_window_size = window_util::DilatedBound( in RewriteInputWithDynamicPadding()
789 window_util::DilatedBound(padded_shape.dimensions(shape_dim), in RewriteInputWithDynamicPadding()
1039 if (window_util::IsTrivialWindowDimension(window_dim)) { in RewriteDynamicReduceWindowSamePadding()
1078 if (window_util::IsTrivialWindowDimension(window_dim)) { in RewriteDynamicSelectAndScatterSamePadding()
1127 const int64 dilated_window_size = window_util::DilatedBound( in RewriteDynamicSelectAndScatterSamePadding()
Dhlo_instructions.cc2144 if (window_util::HasBaseDilation(window)) { in HloConvolutionInstruction()
2147 if (window_util::HasWindowDilation(window)) { in HloConvolutionInstruction()
2156 if (window_util::HasBaseDilation(window())) { in ToCategory()
2159 if (window_util::HasWindowDilation(window())) { in ToCategory()
2180 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2261 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2312 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2452 extra.push_back(StrCat("window={", window_util::ToString(*window_), "}")); in ExtraAttributesToStringImpl()
Dalgebraic_simplifier.cc4798 if (window_util::HasStride(window) || window_util::HasDilation(window) || in HandleReduceWindow()
4799 window_util::HasPadding(window)) { in HandleReduceWindow()
4860 if (HasInteriorPadding(pad_config) && window_util::HasBaseDilation(window)) { in HandleReduceWindow()
4898 if (window_util::HasStride(window)) { in HandleReduceWindow()
4902 if (!window_util::HasSymmetricPadding(pad_config)) { in HandleReduceWindow()
5436 if (window_util::HasStride(window) || window_util::HasPadding(window) || in SimplifyConvToDot()
5437 window_util::HasBaseDilation(window)) { in SimplifyConvToDot()
Ddynamic_dimension_inference.cc732 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleDynamicWindowSamePadding()
1067 if (!window_util::IsTrivialWindowDimension(window_dim)) { in HandleReduceWindow()
/external/tensorflow/tensorflow/lite/micro/examples/micro_speech/
DMakefile.inc71 tensorflow/lite/experimental/microfrontend/lib/window_util.c \
90 tensorflow/lite/experimental/microfrontend/lib/window_util.h \
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD233 "//tensorflow/compiler/xla:window_util",
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1207 if (xla::window_util::HasStride(hlo_reduce_window->window())) { in EmitReduceWindowOp()
1210 if (xla::window_util::HasBaseDilation(hlo_reduce_window->window())) { in EmitReduceWindowOp()
1213 if (xla::window_util::HasWindowDilation(hlo_reduce_window->window())) { in EmitReduceWindowOp()
1218 if (xla::window_util::HasPadding(hlo_reduce_window->window())) { in EmitReduceWindowOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/
DBUILD198 "//tensorflow/compiler/xla:window_util",

12