Searched defs:base_dilations (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | reduce_window_op.cc | 43 std::vector<int64> base_dilations; in Compile() local
|
/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
|
/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
D | xla.py | 353 base_dilations=None, argument
|
/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
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | mhlo_to_lhlo_with_xla.cc | 1196 std::vector<xla::int64> dims, strides, base_dilations, window_dilations, in EmitReduceWindowOp() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 2554 absl::Span<const int64> base_dilations, in ReduceWindowWithGeneralPadding() 2599 absl::Span<const int64> base_dilations, in ReduceWindowInternal() 4644 absl::Span<const int64> base_dilations, in ReduceWindowWithGeneralPadding()
|