Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_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()
Dcudnn_conv_rewriter.cc83 ? !window_util::AllOrNoneReversed(conv->window()) in CanImplementAsCudnnForwardConv()
84 : window_util::HasWindowReversal(conv->window())) { in CanImplementAsCudnnForwardConv()
152 !window_util::HasWindowDilation(conv->window())) { in MatchBackwardFilter()
299 !(window_util::HasBaseDilation(conv->window()) && in MatchBackwardInput()
DBUILD176 "//tensorflow/compiler/xla:window_util",
238 "//tensorflow/compiler/xla:window_util",
430 "//tensorflow/compiler/xla:window_util",
512 "//tensorflow/compiler/xla:window_util",
738 "//tensorflow/compiler/xla:window_util",
756 "//tensorflow/compiler/xla:window_util",
946 "//tensorflow/compiler/xla:window_util",
Dgpu_layout_assignment.cc85 window_util::HasStride(instr->window())) { in HeuristicLayoutAssignment()
Dir_emitter_unnested.cc802 if (window_util::HasDilation(window)) { in HandleSelectAndScatter()
/external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/
DBUILD117 "window_util.c",
121 "window_util.h",
/external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/
DBUILD181 "window_util.cc",
185 "window_util.h",
/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.h24 namespace window_util {
DBUILD728 name = "window_util",
729 srcs = ["window_util.cc"],
730 hdrs = ["window_util.h"],
753 ":window_util",
Dreference_util.cc153 return window_util::StridedBound(unpadded_width, window_len, stride); in WindowCount()
170 window_util::StridedBound(padded_width, window[i], stride[i]); in ReduceWindow1DGeneric()
215 window_util::StridedBound(padded_width, window[i], stride[i]); in ReduceWindow2DGeneric()
324 window_util::StridedBound(padded_width, window[i], stride[i]); in ReduceWindow4DGeneric()
Dwindow_util.cc27 namespace window_util { namespace
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc79 if (window_util::HasWindowReversal(convolution.window())) { in PotentiallyImplementedAsEigenConvolution()
DBUILD282 "//tensorflow/compiler/xla:window_util",
765 "//tensorflow/compiler/xla:window_util",
Dir_emitter.cc757 if (window_util::HasDilation(window)) { in HandleSelectAndScatter()
1032 llvm::ConstantInt* input_bound = b_.getInt64(window_util::DilatedBound( in EmitElementalConvolution()
/external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/
DMakefile.inc69 tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/window_util.cc \
90 tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/window_util.h \
160 tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/window_util.cc
164 tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/window_util.h \
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc360 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleReduceWindow()
381 if (!window_util::IsTrivialWindowDimension(window_dimension)) { in HandleSelectAndScatter()
Dhlo_instructions.cc1865 if (window_util::HasBaseDilation(window)) { in HloConvolutionInstruction()
1868 if (window_util::HasWindowDilation(window)) { in HloConvolutionInstruction()
1877 if (window_util::HasBaseDilation(window())) { in ToCategory()
1880 if (window_util::HasWindowDilation(window())) { in ToCategory()
1901 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
1971 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2018 extra.push_back(StrCat("window={", window_util::ToString(window()), "}")); in ExtraAttributesToStringImpl()
2101 extra.push_back(StrCat("window={", window_util::ToString(*window_), "}")); in ExtraAttributesToStringImpl()
DBUILD198 "//tensorflow/compiler/xla:window_util",
272 "//tensorflow/compiler/xla:window_util",
365 "//tensorflow/compiler/xla:window_util",
539 "//tensorflow/compiler/xla:window_util",
1671 "//tensorflow/compiler/xla:window_util",
1702 "//tensorflow/compiler/xla:window_util",
1738 "//tensorflow/compiler/xla:window_util",
2028 "//tensorflow/compiler/xla:window_util",
2183 "//tensorflow/compiler/xla:window_util",
3290 "//tensorflow/compiler/xla:window_util",
[all …]
Dalgebraic_simplifier.cc3114 if (window_util::HasStride(window) || window_util::HasDilation(window) || in HandleReduceWindow()
3115 window_util::HasPadding(window)) { in HandleReduceWindow()
3167 if (window_util::HasDilation(window)) { in HandleReduceWindow()
3219 if (window_util::HasStride(window)) { in HandleReduceWindow()
3223 if (!window_util::HasSymmetricPadding(pad_config)) { in HandleReduceWindow()
3563 if (window_util::HasStride(window) || window_util::HasPadding(window) || in SimplifyConvToDot()
3564 window_util::HasBaseDilation(window)) { in SimplifyConvToDot()
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_cost_analysis.cc457 window = window_util::MakeWindow({1}); in HandleConvolution()
Dhlo_instruction_test.cc1702 Window w = window_util::MakeWindow({1, 2, 3}); in TEST_F()
1724 Window w = window_util::MakeWindow({1, 2, 3}); in TEST_F()
Dalgebraic_simplifier_test.cc3142 EXPECT_EQ(window_util::ToString(conv->window()), in TEST_P()
3260 EXPECT_EQ(window_util::ToString(conv->window()), in TEST_P()
4232 PaddingConfig padding = window_util::MakeSymmetricPadding( in TEST_P()
4257 Window window = window_util::MakeWindow( in TEST_P()