MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MakeNoPaddingConfig

Function MakeNoPaddingConfig

tensorflow/compiler/xla/util.cc:169–178  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

167}
168
169PaddingConfig MakeNoPaddingConfig(int64 rank) {
170 PaddingConfig padding_config;
171 for (int64 dnum = 0; dnum < rank; ++dnum) {
172 auto dimension = padding_config.add_dimensions();
173 dimension->set_edge_padding_low(0);
174 dimension->set_edge_padding_high(0);
175 dimension->set_interior_padding(0);
176 }
177 return padding_config;
178}
179
180PaddingConfig MakeEdgePaddingConfig(
181 absl::Span<const std::pair<int64, int64>> padding) {

Callers 12

SetMatrixDiagFunction · 0.85
CompileMethod · 0.85
AvgPoolGradFunction · 0.85
TEST_FFunction · 0.85
DiagonalBlocksFunction · 0.85
TEST_PFunction · 0.85
RewriteReductionMethod · 0.85
PadInstructionFunction · 0.85
XLA_TEST_PFunction · 0.85

Calls 1

add_dimensionsMethod · 0.80

Tested by 3

TEST_FFunction · 0.68
TEST_PFunction · 0.68
XLA_TEST_PFunction · 0.68