Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h553 int64 pad_interior[3]; in PadArray3D() local
562 pad_interior[i] = padding.dimensions(i).interior_padding(); in PadArray3D()
565 (input_bounds[i] - 1) * pad_interior[i]; in PadArray3D()
583 if (pad_interior[i] && in PadArray3D()
584 (indices[i] - pad_low[i]) % (pad_interior[i] + 1)) { in PadArray3D()
592 *value = operand((indices[0] - pad_low[0]) / (pad_interior[0] + 1), in PadArray3D()
593 (indices[1] - pad_low[1]) / (pad_interior[1] + 1), in PadArray3D()
594 (indices[2] - pad_low[2]) / (pad_interior[2] + 1)); in PadArray3D()
612 int64 pad_interior[4]; in PadArray4D() local
619 pad_interior[i] = padding.dimensions(i).interior_padding(); in PadArray4D()
[all …]