Searched refs:padding_maps (Results 1 – 5 of 5) sorted by relevance
1056 padding_maps = []1076 padding_maps.append(padding_map)1126 return padded_inputs, padding_maps1362 flat_inputs, padding_maps = _pad_all_input(unpadded_inputs,1365 if padding_maps:1407 for padding_map in padding_maps:
100 repeated tpu.PaddingMap padding_maps = 11; field
342 for (const auto& padding_mapping : metadata_.padding_maps()) { in CompileTFFunctionToHlo()
570 protobuf::RepeatedPtrField<tpu::PaddingMap>* padding_maps) { in FillPaddingMap() argument574 padding_maps->Reserve(padding_map_strs.size()); in FillPaddingMap()576 tpu::PaddingMap* padding_map = padding_maps->Add(); in FillPaddingMap()
643 // padding_maps {672 // CHECK-SAME: padding_maps