Home
last modified time | relevance | path

Searched defs:padding_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc853 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteInputWithDynamicPadding() local
865 auto* padding_dim = padding_configs.mutable_dimensions(shape_dim); in RewriteInputWithDynamicPadding() local
1225 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteDynamicSelectAndScatterSamePadding() local
Dhlo_parser.cc5089 std::vector<int64> padding_dim; in ParsePaddingConfig() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc890 int64_t padding_dim = slice_sharding_dim; in PartitionBaseCase() local