[Mosaic][SC] Add custom assembly format to tpu.enqueue_indirect_dma

PiperOrigin-RevId: 786844207

Author: naummo

jaxlib/mosaic/dialect/tpu/tpu.td

@@ -903,6 +903,9 @@ def TPU_EnqueueIndirectDMAOp : TPU_Op<"enqueue_indirect_dma">, IndirectDMAOp {
     Optional<I32>:$offset_filter,
     DefaultValuedAttr<BoolAttr, "false">:$add
   );
+  let assemblyFormat = [{
+    $source `to` $target `at` $offsets `,` `semaphore` $semaphore (`offset_filter` $offset_filter^)? attr-dict `:` type($source) `,` type($target) `,` type($offsets) `,` type($semaphore)
+    }];
   let hasVerifier = 1;
   let extraClassDeclaration = extraBaseClassDeclaration # [{
     LogicalResult verifyGather(MemRefType operand_ty,