Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.cc159 int64_t pad_low = padding[0].first; in ReduceWindow1DGeneric() local
195 std::vector<int64_t> pad_low(window.size(), 0); in ReduceWindow3DAdd() local
255 std::vector<int64_t> pad_low(window.size(), 0); in ReduceWindow4DGeneric() local
341 std::vector<int64_t> pad_low(window.size(), 0); in SelectAndScatter4DGePlus() local
H A Dreference_util.h556 int64_t pad_low[3]; in PadArray3D() local
615 int64_t pad_low[4]; in PadArray4D() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_window_test.cc580 int64_t pad_low[4]; member
1102 int64_t pad_low[2]; member
1341 int64_t pad_low[1]; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dspace_to_batch_converter.cc1389 const int64_t pad_low = in CanPropagate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc1777 int64_t pad_low; in inferConvReturnTypeComponents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc1181 Value pad_low, pad_high; in matchAndRewriteDynamicConv() local
1331 int64_t pad_low, pad_high; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1192 int64_t pad_low; in ComputeConvWindowedOutputSize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc4153 int64_t pad_low = hlo->window().dimensions(i).padding_low(); in HandleSelectAndScatter() local