Home
last modified time | relevance | path

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

/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 {
DBUILD774 name = "window_util",
775 srcs = ["window_util.cc"],
776 hdrs = ["window_util.h"],
793 ":window_util",
809 ":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.cc83 ? !window_util::AllOrNoneReversed(conv->window()) in CanImplementAsGpuForwardConv()
84 : window_util::HasWindowReversal(conv->window())) { in CanImplementAsGpuForwardConv()
150 !window_util::HasWindowDilation(conv->window())) { in MatchBackwardFilter()
379 !(window_util::HasBaseDilation(conv->window()) && in MatchBackwardInput()
DBUILD267 "//tensorflow/compiler/xla:window_util",
330 "//tensorflow/compiler/xla:window_util",
598 "//tensorflow/compiler/xla:window_util",
743 "//tensorflow/compiler/xla:window_util",
986 "//tensorflow/compiler/xla:window_util",
1005 "//tensorflow/compiler/xla:window_util",
1035 "//tensorflow/compiler/xla:window_util",
1356 "//tensorflow/compiler/xla:window_util",
1755 "//tensorflow/compiler/xla:window_util",
Dgpu_layout_assignment.cc101 window_util::HasStride(instr->window())) { in HeuristicLayoutAssignment()
Dir_emitter_unnested.cc623 if (window_util::HasDilation(window)) { in HandleSelectAndScatter()
/external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/
DBUILD118 "window_util.c",
122 "window_util.h",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc79 if (window_util::HasWindowReversal(convolution.window())) { in PotentiallyImplementedAsEigenConvolution()
DBUILD289 "//tensorflow/compiler/xla:window_util",
765 "//tensorflow/compiler/xla:window_util",
Dir_emitter.cc817 if (window_util::HasDilation(window)) { in HandleSelectAndScatter()
1095 llvm::ConstantInt* input_bound = b_.getInt64(window_util::DilatedBound( in EmitElementalConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/
DBUILD30 "//tensorflow/compiler/xla:window_util",
Dconv_emitter.cc767 if (window_util::HasWindowReversal(conv->window())) { in ConvIsImplemented()
770 if (window_util::HasDilation(conv->window())) { in ConvIsImplemented()
/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/service/
DBUILD221 "//tensorflow/compiler/xla:window_util",
296 "//tensorflow/compiler/xla:window_util",
447 "//tensorflow/compiler/xla:window_util",
640 "//tensorflow/compiler/xla:window_util",
1870 "//tensorflow/compiler/xla:window_util",
1903 "//tensorflow/compiler/xla:window_util",
1938 "//tensorflow/compiler/xla:window_util",
1977 "//tensorflow/compiler/xla:window_util",
2012 "//tensorflow/compiler/xla:window_util",
2361 "//tensorflow/compiler/xla:window_util",
[all …]
Dhlo_instructions.cc1988 if (window_util::HasBaseDilation(window)) { in HloConvolutionInstruction()
1991 if (window_util::HasWindowDilation(window)) { in HloConvolutionInstruction()
2000 if (window_util::HasBaseDilation(window())) { in ToCategory()
2003 if (window_util::HasWindowDilation(window())) { in ToCategory()
2024 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2094 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2141 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2226 extra.push_back(StrCat("window={", window_util::ToString(*window_), "}")); in ExtraAttributesToStringImpl()
Dalgebraic_simplifier.cc3756 if (window_util::HasStride(window) || window_util::HasDilation(window) || in HandleReduceWindow()
3757 window_util::HasPadding(window)) { in HandleReduceWindow()
3818 if (HasInteriorPadding(pad_config) && window_util::HasBaseDilation(window)) { in HandleReduceWindow()
3856 if (window_util::HasStride(window)) { in HandleReduceWindow()
3860 if (!window_util::HasSymmetricPadding(pad_config)) { in HandleReduceWindow()
4268 if (window_util::HasStride(window) || window_util::HasPadding(window) || in SimplifyConvToDot()
4269 window_util::HasBaseDilation(window)) { in SimplifyConvToDot()
Ddynamic_dimension_inference.cc922 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleReduceWindow()
945 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleSelectAndScatter()
Dshape_inference.cc201 !window_util::IsTrivialWindowDimension(dim)) { in InferWindowOutputShape()
204 window_util::ToString(window)); in InferWindowOutputShape()
207 const int64 dilated_base = window_util::DilatedBound( in InferWindowOutputShape()
212 window_util::DilatedBound(dim.size(), dim.window_dilation()); in InferWindowOutputShape()
214 output_dimensions[i] = window_util::StridedBound( in InferWindowOutputShape()
Dhlo_instruction_test.cc1777 Window w = window_util::MakeWindow({1, 2, 3}); in TEST_F()
1811 Window w = window_util::MakeWindow({1, 2, 3}); in TEST_F()
Dhlo_cost_analysis.cc555 window = window_util::MakeWindow({1}); in HandleConvolution()
Dalgebraic_simplifier_test.cc3528 EXPECT_EQ(window_util::ToString(conv->window()), in TEST_P()
3646 EXPECT_EQ(window_util::ToString(conv->window()), in TEST_P()
4642 PaddingConfig padding = window_util::MakeSymmetricPadding( in TEST_P()
4667 Window window = window_util::MakeWindow( in TEST_P()