Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_simplify_padding.cc97 const PaddingConfig::PaddingConfigDimension& padding_config = in NumTrailingZeroOutputFeatures()
144 const PaddingConfig::PaddingConfigDimension& padding_config = in NumTrailingZeroOutputFeatures()
371 PaddingConfig::PaddingConfigDimension* new_pad_feature_dim = in TrySimplifyPadding()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.pyi104 class PaddingConfigDimension:
111 dimensions: List[PaddingConfigDimension]
Dxla_client.py418 class PaddingConfigDimension: class
457 dimension = PaddingConfigDimension()
Dtypes.h378 xla::PaddingConfig::PaddingConfigDimension* config_dim =
Dxla_client_test.py1300 dimension = xla_client.PaddingConfigDimension()
/external/tensorflow/tensorflow/compiler/xla/
Dwindow_util.cc164 [](const PaddingConfig::PaddingConfigDimension& dim) { in HasSymmetricPadding()
Dxla_data.proto96 message PaddingConfigDimension { message
109 repeated PaddingConfigDimension dimensions = 1;
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_normalization.cc433 bool IsZeroPadding(const PaddingConfig::PaddingConfigDimension& c) { in IsZeroPadding()
Dhlo_creation_utils.cc657 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros()
Ddynamic_padder.cc908 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteInputWithDynamicPadding()
1275 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteDynamicSelectAndScatterSamePadding()
Dhlo_instruction.cc4073 [](const PaddingConfig::PaddingConfigDimension& dim) { in PaddingConfigToString()
4078 [&](std::string* out, const PaddingConfig::PaddingConfigDimension& dim) { in PaddingConfigToString()
Dshape_inference.cc550 [](const PaddingConfig::PaddingConfigDimension& p) { in InferPadShape()
Dalgebraic_simplifier.cc3843 PaddingConfig::PaddingConfigDimension* padding_dimension = in HandlePad()
3869 const PaddingConfig::PaddingConfigDimension& padding_dimension = in HandlePad()
Dalgebraic_simplifier_test.cc2257 PaddingConfig::PaddingConfigDimension* dimension = padding.add_dimensions(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1890 `PaddingConfig` is a repeated field of `PaddingConfigDimension`, which contains