1. 12 10月, 2019 3 次提交
  2. 11 10月, 2019 11 次提交
  3. 10 10月, 2019 5 次提交
    • H
      Internal change · cf4cae61
      Hongkun Yu 提交于
      PiperOrigin-RevId: 273966871
      cf4cae61
    • H
      experimental_allow_stateful is not necessary with make dataset fn · 8110bb64
      Hongkun Yu 提交于
      PiperOrigin-RevId: 273861263
      8110bb64
    • R
      Simply LayerNorm mixed precision logic. · 0257b276
      Reed Wanderman-Milne 提交于
      Instead of needing to ensure variables are float32, casting inputs to float32, etc, instead dtype="float32" is passed to the layer constructor, which will do all that logic automatically.
      
      The only difference is the output of LayerNorm is now float32 instead of float16, so an extra cast is needed elsewhere.
      
      PiperOrigin-RevId: 273833286
      0257b276
    • P
      Add Combined NMS (#6138) · 3980d2a1
      Pooya Davoodi 提交于
      * Updating python API to use CombinedNonMaxSuppresion TF operator
      
      1. Adds a unit test to test post_processing python API
      2. Currently sets clip_window to None as the kernel uses the default
         clip_window of [0,0,1,1]
      3. Added use_static_shapes to the API. In old API if
         use_static_shapes is true, then it pads/clips outputs to max_total_size, if
      specified. If not specified, it pads to num_classes*max_size_per_class.
       If use_static_shapes is false, it always pads/clips to max_total_size.
      
      Update unit test to account for clipped bouding boxes
      
      Changed the name to CombinedNonMaxSuppression based on feedback from Google
      
      Added additional parameters to combinedNMS python function. They are currently
      unused and required for networks like FasterRCNN and MaskRCNN
      
      * Delete selected_indices from API
      
      Because it was removed from CombinedNMS recently in the PR.
      
      * Improve doc of function combined_non_max_suppression
      
      * Enable CombinedNonMaxSuppression for first_stage_nms
      
      * fix bug
      
      * Ensure agnostic_nms is not used with combined_nms
      
      Remove redundant arguments from combined_nms
      
      * Fix pylint
      
      * Add checks for unsupported args
      
      * Fix pylint
      
      * Move combined_non_max_suppression to batch_multiclass_non_max_suppression
      
      Also rename combined_nms to use_combined_nms
      
      * Delete combined_nms for first_stage_nms because it does not work
      
      * Revert "Delete combined_nms for first_stage_nms because it does not work"
      
      This reverts commit 2a3cc5145f17cee630a67ddedd20e90c2920fa9f.
      
      * Use nmsed_additional_fields.get to avoid error
      
      * Merge combined_non_max_suppression with main nms function
      
      * Rename combined_nms for first stage nms
      
      * Improve  docs
      
      * Use assertListEqual for numpy arrays
      
      * Fix pylint errors
      
      * End comments with period
      3980d2a1
    • A
      Fix BERT benchmark test. · f5e2211f
      A. Unique TensorFlower 提交于
      PiperOrigin-RevId: 273795511
      f5e2211f
  4. 09 10月, 2019 2 次提交
  5. 08 10月, 2019 4 次提交
  6. 07 10月, 2019 1 次提交
  7. 06 10月, 2019 1 次提交
  8. 05 10月, 2019 2 次提交
  9. 04 10月, 2019 3 次提交
  10. 03 10月, 2019 2 次提交
  11. 02 10月, 2019 1 次提交
  12. 01 10月, 2019 4 次提交
  13. 30 9月, 2019 1 次提交