Home
last modified time | relevance | path

Searched defs:pad_low (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc158 int64 pad_low = padding[0].first; in ReduceWindow1DGeneric() local
194 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow3DAdd() local
254 std::vector<int64> pad_low(window.size(), 0); in ReduceWindow4DGeneric() local
339 std::vector<int64> pad_low(window.size(), 0); in SelectAndScatter4DGePlus() local
Dreference_util.h543 int64 pad_low[3]; in PadArray3D() local
602 int64 pad_low[4]; in PadArray4D() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc582 int64 pad_low[4]; member
1077 int64 pad_low[2]; member
1292 int64 pad_low[1]; member
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc116 auto pad_low = padding.template getValue<llvm::APInt>({i, 0}); in MapWindowIvsToInput() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc981 const int64 pad_low = in CanPropagate() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1653 tensorflow::int64 pad_low; in inferConvReturnTypes() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3344 int64 pad_low = hlo->window().dimensions(i).padding_low(); in HandleSelectAndScatter() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1211 int64_t pad_low, pad_high; in matchAndRewrite() local