HomeSort by relevance Sort by last modified time
    Searched refs:window_dim (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 91 for (const WindowDimension& window_dim : conv->window().dimensions()) {
92 if (window_dim.stride() != 1) {
98 if (window_dim.base_dilation() != 1) {
104 if (window_dim.padding_low() < 0) {
108 if (window_dim.window_reversal()) {
255 for (const WindowDimension& window_dim : conv->window().dimensions()) {
256 if (window_dim.stride() != 1) {
262 if (window_dim.window_dilation() != 1) {
268 if (window_dim.window_reversal()) {
cudnn_convolution_rewriter_test.cc 40 WindowDimension* window_dim = default_conv_window_.add_dimensions(); local
41 window_dim->set_size(1);
42 window_dim->set_stride(1);
43 window_dim->set_padding_low(0);
44 window_dim->set_padding_high(0);
45 window_dim->set_window_dilation(1);
46 window_dim->set_base_dilation(1);
291 const WindowDimension& window_dim = custom_call->window().dimensions(i); local
294 EXPECT_EQ(3, window_dim.padding_low());
295 EXPECT_EQ(3, window_dim.padding_high())
420 const WindowDimension& window_dim = custom_call->window().dimensions(i); local
    [all...]
pad_insertion.cc 191 void IncreasePaddingLowBy(int64 delta, WindowDimension* window_dim) {
192 window_dim->set_padding_low(window_dim->padding_low() + delta);
195 void IncreasePaddingHighBy(int64 delta, WindowDimension* window_dim) {
196 window_dim->set_padding_high(window_dim->padding_high() + delta);
  /external/tensorflow/tensorflow/compiler/xla/
window_util.cc 194 const WindowDimension& window_dim = window.dimensions(logical_dim);
195 return window_dim.size() == 1 && window_dim.stride() == 1 &&
196 window_dim.padding_low() == 0 && window_dim.padding_high() == 0;
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator.cc     [all...]
algebraic_simplifier.cc 1812 auto& window_dim = *new_window.mutable_dimensions(i); local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc     [all...]

Completed in 661 milliseconds