Commits

Branch selector

 master

User selector

gnecula

All time

Commit History

Commits on May 31, 2024

• Add DynamicApproxTopKOp to StableHLO

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 31, 2024
  d079b64

Commits on May 2, 2024

• [StablehloShapeRefinement] Skip constant folding of convert operations with dynamic shapes.

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 2, 2024
  f2f2f96

Commits on Jan 13, 2024

• [XlaCallModule] Fix multi-platform support.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJan 13, 2024
  6d618b1

Commits on Jan 11, 2024

• [XlaCallModule] Fixes for serialization version 9.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJan 11, 2024
  92d5caa

Commits on Dec 7, 2023

• Add flatbuffers as a test dependency for JAX

  

  gnecula

  authored and

  tensorflower-gardener

  committedDec 7, 2023
  27acc5e

Commits on Oct 27, 2023

• [XlaCallModule] Improved support for effects and version 9.

  

  gnecula

  authored and

  tensorflower-gardener

  committedOct 27, 2023
  c9b4c57

Commits on Oct 23, 2023

• [XlaCallModule] Add more logging.

  

  gnecula

  authored and

  tensorflower-gardener

  committedOct 23, 2023
  c251034

Commits on Oct 20, 2023

• [XlaCallModule] Add back support for version 4.

  

  gnecula

  authored and

  tensorflower-gardener

  committedOct 20, 2023
  2260565

Commits on Oct 18, 2023

• [XlaCallModule] Drop support for dim_args_spec attribute.

  

  gnecula

  authored and

  tensorflower-gardener

  committedOct 18, 2023
  9377899

• [XlaCallModule] Allow i64 platform index arguments.

  

  gnecula

  authored and

  tensorflower-gardener

  committedOct 18, 2023
  403c8c2

Commits on Oct 16, 2023

• [XlaCallModule] Allow i64 platform index arguments.

  

  gnecula

  authored and

  tensorflower-gardener

  committedOct 16, 2023
  2758a19

Commits on Sep 21, 2023

• Improve shape refinement to not require inlining.

  

  gnecula

  authored and

  tensorflower-gardener

  committedSep 21, 2023
  060f511

• Improve shape refinement to not require inlining.

  

  gnecula

  authored and

  tensorflower-gardener

  committedSep 21, 2023
  0c7098d

Commits on Aug 31, 2023

• Add XlaCallModule test for multi-platform lowering.

  

  gnecula

  authored and

  tensorflower-gardener

  committedAug 31, 2023
  253ac98

Commits on Aug 29, 2023

• Add tests and cleanup for multi-platform support in XlaCallModule.

  

  gnecula

  authored and

  tensorflower-gardener

  committedAug 29, 2023
  941af58

Commits on Aug 7, 2023

• Remove one use of inlining in XlaCallModule shape refinement.

  

  gnecula

  authored and

  tensorflower-gardener

  committedAug 7, 2023
  1ccacdd

Commits on Jul 28, 2023

• Introduce the DynamicTopKOp experiment

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 28, 2023
  0f3da2e

• XlaCallModule: Increase limit of number of error message inputs for shape assertions

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 28, 2023
  f374749

Commits on Jul 26, 2023

• Remove one use of inlining in XlaCallModule shape refinement.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 26, 2023
  35b4081

Commits on Jul 21, 2023

• Introduce version 8 of XlaCallModule.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 21, 2023
  9699c77

Commits on Jul 14, 2023

• Separates ValidateStaticShapes from RefineDynamicShapes.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 14, 2023
  78f425b

Commits on Jul 12, 2023

• Add support for shape assertions to RefinePolymorphicShapes

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 12, 2023
  4b81db9

• Refactor xla_call_module_loader to share refine_polymorphic_shapes

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 12, 2023
  21eb514

Commits on Jul 11, 2023

• Improve DumpMlirOpToFile when dumping to stderr.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 11, 2023
  52a0ee8

Commits on Jul 5, 2023

• Simplify XlaCallModule test for disabling platform check.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJul 5, 2023
  3ba0e0e

Commits on Jun 27, 2023

• [shape_poly] Shape polymorphism support for approx_top_k

  

  gnecula

  authored and

  tensorflower-gardener

  committedJun 27, 2023
  9e74ca4

Commits on Jun 22, 2023

• Documentation improvements for XlaCallModule.disabled_checks.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJun 22, 2023
  e941e39

Commits on Jun 8, 2023

• Introduce a tf_xla_call_module_disabled_checks flag for controlling XlaCallModule safety checks.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJun 8, 2023
  ff4139f

Commits on Jun 7, 2023

• Add disabled_checks attribute to XlaCallModule.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJun 7, 2023
  d592d80

Commits on Jun 3, 2023

• [shape_poly] Add a polymorphic shape refinement MLIR pass accessible to JAX Python.

  

  gnecula

  authored and

  tensorflower-gardener

  committedJun 3, 2023
  9f509d0

Commits on May 26, 2023

• Remove the dim_args_spec from the XlaCallModule tests.

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 26, 2023
  a3b84b1

Commits on May 24, 2023

• Stop supporting XlaCallModule without compatibility guarantees.

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 24, 2023
  2d3de03

• Replace tsl errors with absl errors.

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 24, 2023
  a935a88

Commits on May 18, 2023

• Enforce that the actual arguments for XlaCallModule match the expected types and shapes.

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 18, 2023
  00444b5

Commits on May 15, 2023

• Improve logging for JAX native serialization.

  

  gnecula

  authored and

  tensorflower-gardener

  committedMay 15, 2023
  b383ebd

Pagination

PreviousNext