Home
last modified time | relevance | path

Searched defs:base_dilations (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc43 std::vector<int64> base_dilations; in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc657 llvm::SmallVector<int64, 4> sizes, strides, base_dilations, win_dilations; in ImportInstructionImpl() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py353 base_dilations=None, argument
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc177 llvm::SmallVector<int64, 4> sizes, strides, base_dilations, win_dilations; in ReduceWindowInternal() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1196 std::vector<xla::int64> dims, strides, base_dilations, window_dilations, in EmitReduceWindowOp() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2554 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()