Home
last modified time | relevance | path

Searched refs:pad_interior (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h545 int64 pad_interior[3]; in PadArray3D() local
554 pad_interior[i] = padding.dimensions(i).interior_padding(); in PadArray3D()
557 (input_bounds[i] - 1) * pad_interior[i]; in PadArray3D()
575 if (pad_interior[i] && in PadArray3D()
576 (indices[i] - pad_low[i]) % (pad_interior[i] + 1)) { in PadArray3D()
584 *value = operand((indices[0] - pad_low[0]) / (pad_interior[0] + 1), in PadArray3D()
585 (indices[1] - pad_low[1]) / (pad_interior[1] + 1), in PadArray3D()
586 (indices[2] - pad_low[2]) / (pad_interior[2] + 1)); in PadArray3D()
604 int64 pad_interior[4]; in PadArray4D() local
611 pad_interior[i] = padding.dimensions(i).interior_padding(); in PadArray4D()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo_patterns.td225 $pad_interior),
227 [(IsZero $pad_interior)]>;