Searched refs:win_dilations (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 177 llvm::SmallVector<int64, 4> sizes, strides, base_dilations, win_dilations; in ReduceWindowInternal() local
183 win_dilations.push_back(dim.window_dilation()); in ReduceWindowInternal()
195 GetI64ElementsAttr(win_dilations, &builder_), in ReduceWindowInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 657 llvm::SmallVector<int64, 4> sizes, strides, base_dilations, win_dilations; in ImportInstructionImpl() local
663 win_dilations.push_back(dim.window_dilation()); in ImportInstructionImpl()
674 "window_dilations", ConvertDimensions(win_dilations))); in ImportInstructionImpl()
|