Home
last modified time | relevance | path

Searched refs:base_dilation (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc64 if (dim.base_dilation() != 1) { in ToString()
65 StrAppend(&str, ",base_dilation=", dim.base_dilation()); in ToString()
106 return StrCat(dim.base_dilation()); in ToString()
162 if (dim.base_dilation() != 1) { in HasBaseDilation()
212 window_dimension.base_dilation() == 1; in IsTrivialWindowDimension()
Dxla_data.proto415 // means no dilation. base_dilation - 1 no-op entries ("holes") are implicitly
418 int64 base_dilation = 6; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc506 window_dim.base_dilation() > 1 in HandleConvolution()
507 ? undilated_index / window_dim.base_dilation() in HandleConvolution()
511 if (undilated_index != lhs_spatial_index * window_dim.base_dilation()) { in HandleConvolution()
Dhlo_evaluator_typed_visitor.h1130 if (window_dim.base_dilation() > 1 && in HandleConvolution()
1131 undilated_index % window_dim.base_dilation() != 0) { in HandleConvolution()
1138 if (window_dim.base_dilation() > 1) { in HandleConvolution()
1139 lhs_spatial_index = undilated_index / window_dim.base_dilation(); in HandleConvolution()
2839 if (base_index[i] % window.dimensions(i).base_dilation() != 0) { in IterateThroughWindow()
2844 base_index[i] /= window.dimensions(i).base_dilation(); in IterateThroughWindow()
Dalgebraic_simplifier.cc3435 w.base_dilation() != 1)) { in FoldConvInputPad()
3438 if (w.base_dilation() != 1 && in FoldConvInputPad()
3447 CHECK_EQ(w.base_dilation(), 1); in FoldConvInputPad()
Dshape_inference.cc189 if (dim.base_dilation() < 1) { in InferWindowOutputShape()
208 ShapeUtil::GetDimension(base_shape, i), dim.base_dilation()); in InferWindowOutputShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc76 if (conv_window->dimensions(i).base_dilation() != 1) { in MaybePaddedAndSlicedInput()
78 conv_window->dimensions(i).base_dilation() - 1); in MaybePaddedAndSlicedInput()
Dcudnn_conv_rewriter.cc135 if (window_dim.base_dilation() != 1) { in MatchBackwardFilter()
343 dim->set_stride(old_window.dimensions(i).base_dilation()); in MatchBackwardInput()
Delemental_ir_emitter.cc403 index_typed_const(window.dimensions(i).base_dilation())), in MakeElementGenerator()
410 index_typed_const(window.dimensions(i).base_dilation())); in MakeElementGenerator()
Dcudnn_conv_rewriter_test.cc318 EXPECT_EQ(1, window_dim.base_dilation()); in TEST_F()
451 EXPECT_EQ(1, window_dim.base_dilation()); in TEST_F()
Dcudnn_conv_runner.cc162 CHECK_EQ(dim.base_dilation(), 1) in RunCudnnConvImpl()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc682 b_.getInt64(window.dimensions(i).base_dilation())), in EmitElementalReduceWindow()
693 b_.getInt64(window.dimensions(i).base_dilation())); in EmitElementalReduceWindow()
1025 const auto not_in_hole = [&](llvm::Value* input_index, int64 base_dilation) { in EmitElementalConvolution() argument
1026 llvm::Value* remainder = SRem(input_index, b_.getInt64(base_dilation)); in EmitElementalConvolution()
1034 window.dimensions(i).base_dilation())); in EmitElementalConvolution()
1037 not_in_hole(input_spatial[i], window.dimensions(i).base_dilation()); in EmitElementalConvolution()
1044 const auto undilate = [&](llvm::Value* input_index, int64 base_dilation) { in EmitElementalConvolution() argument
1045 return SDiv(input_index, b_.getInt64(base_dilation)); in EmitElementalConvolution()
1049 undilate(input_spatial[i], window.dimensions(i).base_dilation()); in EmitElementalConvolution()
1170 int64 lhs_row_dilation = window.dimensions(0).base_dilation(); in HandleConvolution()
[all …]